Update app.py

app.py

@@ -9,7 +9,7 @@ class YOLODetect():
     def __init__(self, modelo):
         self.modelo = modelo
 
-    def predecir(self, source, imgsz=1280, conf=0.7, iou=0.50):
+    def predecir(self, source, imgsz=640, conf=0.7, iou=0.50):
         self.results = self.modelo.predict(source=source, save=True, imgsz=imgsz, conf=conf, iou=iou)
         return self.results
 
@@ -57,7 +57,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         iou_threshold = gr.Slider(label="NMS IoU Threshold (0.0 - 1.0)", minimum=0.0, maximum=1.0, value=0.8)
         conf_threshold = gr.Slider(label="Umbral o threshold (0.0 - 1.0)", minimum=0.0, maximum=1.0, value=0.9)
-        size = gr.Dropdown(label="Tamaño de la imagen", choices=["640", "1280"], value="1280")
+        size = gr.Dropdown(label="Tamaño de la imagen", choices=["640", "1280"], value="640")
 
     with gr.Row():
         uploaded_image = gr.Image(type="pil", label="Cargar imagen")