Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 from ultralytics import YOLO
 from PIL import Image
 
-# We laden de modellen in een dictionary zodat ze maar één keer geladen worden
 models = {
-    "CowCatcherV15" : YOLO('cowcatcherV15.pt'),
-    "CowCatcherV16 (Experimental)": YOLO('cowcatcherV16.pt') # Pas de bestandsnaam aan
+    "CowCatcherV15": YOLO('cowcatcherV15.pt'),
+    "CowCatcherV16 (Experimental)": YOLO('cowcatcherV16.pt')
 }
 
 description_text = """
@@ -14,23 +13,18 @@ By analyzing live footage from your barn cameras, it automatically detects "moun
 """
 
 def predict(img, model_name, conf_threshold):
-    # Selecteer het juiste model uit de dictionary
     selected_model = models[model_name]
-    
-    # Voer de voorspelling uit met de gekozen confidence
     results = selected_model(img, conf=conf_threshold)
     res_plotted = results[0].plot()
-    
     return Image.fromarray(res_plotted[:, :, ::-1])
 
-# De Interface
 demo = gr.Interface(
     fn=predict,
     inputs=[
         gr.Image(type="pil", label="Upload barn image"),
         gr.Dropdown(
             choices=list(models.keys()), 
-            value="CowCatcher v1 (Standard)", 
+            value="CowCatcherV15", 
             label="Select Model"
         ),
         gr.Slider(