jeffaudi commited on
Commit
f13b6ca
·
1 Parent(s): 95787c5

Corrected function call

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ LINE_WIDTH = 2
28
  logger.info(f"Gradio version: {gr.__version__}")
29
 
30
  # Define the inference function
31
- def predict_image(image, threshold):
32
 
33
  # Resize the image to the new size
34
  #image = image.resize((image.size[0] * 2, image.size[1] * 2))
 
28
  logger.info(f"Gradio version: {gr.__version__}")
29
 
30
  # Define the inference function
31
+ def predict_image(image, text_prompt, box_threshold, text_threshold):
32
 
33
  # Resize the image to the new size
34
  #image = image.resize((image.size[0] * 2, image.size[1] * 2))