Update deploy.py

deploy.py

@@ -3,14 +3,14 @@ from faster_rcnn import img_detect, video_detection
 
 
 # Define detection function
-def detect(choice, input):
+def detect(choice, inp):
     # Call the appropriate detection function based on the choice
     if choice == "Image":
-        return img_detect(input)
+        return img_detect(inp)
     elif choice == "Video":
-        return video_detection(input)
+        return video_detection(inp)
     else:
-        return img_detect(input)  # Default to image detection
+        return img_detect(inp)  # Default to image detection
 
 
 # Create Gradio interface using blocks layout
@@ -19,7 +19,7 @@ with gr.Blocks() as demo:
         choices=["Image", "Video"],
         label="What type of object would you like to detect?",
     )
-    inp_image = gr.Image(filepath="image")
+    inp_image = gr.Image(filepath="upload")
     output = gr.Image()
     det = gr.Button('Detect')
     det.click(detect, [choices, inp_image], output)