Hugging Face's logo

Spaces:
camenduru
/
birefnet
Runtime error

App Files Community
camenduru commited on
Commit
956eed8
•
1 Parent(s): d87723f

Update worker_runpod.py

Browse files
Files changed (1) hide show
  1. worker_runpod.py +1 -1
worker_runpod.py CHANGED
@@ -30,7 +30,7 @@ with torch.inference_mode():
30
  def generate(input):
31
  values = input["input"]
32
 
33
- input_image_url = values['input_image_url']
34
  input_image = download_file(input_image_url)
35
 
36
  input_image = Image.open(input_image)
 
30
  def generate(input):
31
  values = input["input"]
32
 
33
+ input_image_url = values['input_image_check']
34
  input_image = download_file(input_image_url)
35
 
36
  input_image = Image.open(input_image)