Change the label

app.py

@@ -17,9 +17,9 @@ def load_model(model_path):
 detection_model = load_model(model_path)
 
 # Function to perform prediction on an image
-def predict_image(input_img):
+def predict_image(Input_Image):
     try:
-        result = detection_model.predict(input_img, conf=0.5, iou=0.6)
+        result = detection_model.predict(Input_Image, conf=0.5, iou=0.6)
         img_bgr = result[0].plot()
         out_pilimg = Image.fromarray(img_bgr[..., ::-1])  # Convert BGR to RGB for PIL image
         return out_pilimg
@@ -27,9 +27,9 @@ def predict_image(input_img):
         return f"Error processing image: {str(e)}"
 
 # Function to perform prediction on a video
-def predict_video(video_path):
+def predict_video(Input_Video):
     try:
-        result_video = detection_model.predict(video_path, conf=0.5, iou=0.6, save=True)
+        result_video = detection_model.predict(Input_Video, conf=0.5, iou=0.6, save=True)
         
         # Find the processed video output path
         processed_video_path = str(result_video[0].save_dir / result_video[0].path.name)
@@ -41,7 +41,7 @@ def predict_video(video_path):
 image_interface = gr.Interface(
     fn=predict_image, 
     inputs=gr.Image(type="pil"), 
-    outputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil", label="Processed_Image"),
     title="Object Detection - Image",
     description="Upload an image and the model will detect objects."
 )
@@ -49,7 +49,7 @@ image_interface = gr.Interface(
 video_interface = gr.Interface(
     fn=predict_video, 
     inputs=gr.Video(), 
-    outputs=gr.File(label="Processed Video"),
+    outputs=gr.File(label="Processed_Video"),
     title="Object Detection - Video",
     description="Upload a video and the model will detect objects."
 )