App components update

app.py

@@ -26,7 +26,7 @@ for i, url in enumerate(file_urls):
 model = YOLO('best.pt')
 path  = [['image_0.jpg'], ['image_1.jpg']]
 
-def show_preds(image_path):
+def show_preds(image_path, video_path=None):
     image = cv2.imread(image_path)
     outputs = model.predict(source=image_path, return_outputs=True)
     for image_id, result in enumerate(outputs):
@@ -43,14 +43,21 @@ def show_preds(image_path):
             )
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
-gr_interface = gr.Interface(
+inputs_image = [
+    gr.components.Image(type="filepath", label="Input Image"),
+    # gr.components.Video(type="filepath", label="Input Video", optional=True),
+
+]
+outputs_image = [
+    gr.components.Image(type="numpy", label="Output Image"),
+]
+interface = gr.Interface(
     fn=show_preds,
-    inputs=gr.inputs.Image(type="filepath", label="Input Image"),
-    outputs=gr.outputs.Image(type="numpy", label="Output Image"),
+    inputs=inputs_image,
+    outputs=outputs_image,
     title="Pothole detector",
     examples=path,
     cache_examples=False,
     # live=True,
 )
-
-gr_interface.launch(debug=True, enable_queue=True)
+interface.launch(debug=True, enable_queue=True)