Update app.py

app.py

@@ -76,11 +76,11 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
         # Assuming output_path is the path to the output file
     _, output_extension = os.path.splitext(output_path)
     if output_extension.lower() in img_extensions:
-        output = output_path  # Load the image file here
+        output_image = output_path  # Load the image file here
     elif output_extension.lower() in vid_extensions:
-        output = output_path  # Load the video file here
+        output_video = output_path  # Load the video file here
 
-    return output, output_path
+    return output_image, output_video, output_path
 
 
 def app():
@@ -113,12 +113,8 @@ def app():
                 yolov9_infer = gr.Button(value="Inference")
             with gr.Column():
                 gr.HTML("<h2>Output</h2>")
-                if img_path is not None:
-                    output_image = gr.Image(type="numpy",label="Output")
-                    output = output_image
-                else:
-                    output_video = gr.Video(label="Output")
-                    output = output_video
+                output_image = gr.Image(type="numpy",label="Output")
+                output_video = gr.Video(label="Output")
                 output_path = gr.Textbox(label="Output path")
 
         yolov9_infer.click(
@@ -129,7 +125,7 @@ def app():
                 vid_path,
                 tracking_algorithm
             ],
-            outputs=[output, output_path],
+            outputs=[output_image, output_video, output_path],
         )