Update app.py

app.py

@@ -6,17 +6,16 @@ import gradio as gr
 from ultralytics import YOLO
 
 
-def predict(path:str):
+def predict(path):
     model = YOLO("yolov8s.yaml")
     model = YOLO("best.pt")
-    #imagen = cv2.imread(path)
     results = model.predict(source=path)
 
     for r in results:
-        return r.plot(show_conf=True, line_width=None, font_size=None, pil=False, example='abc')
+        return r.plot(show_conf=True, line_width=None, font_size=None, pil=False)
 
 
-gr.Interface(title='Generación 16, Módulo 4',
+gr.Interface(title='Generación 16 - Módulo 4',
              fn=predict,
              inputs=gr.components.Image(type="filepath", label="Input"),
              outputs=gr.components.Image(type="numpy", label="Output")).launch()