Update app.py

app.py

@@ -25,13 +25,15 @@ def make_prediction(img, feature_extractor, model):
 
 
 def detect_objects(experiment_type, attention_heads_num, image_input, threshold, display_mask=False, img_input_mask=None):
-    if experiment_type == "REPRODUCABILITY CHECK 1": 
+    if experiment_type == "Parameters verification": 
         model_repo = f"polejowska/detr-r50-cd45rb-all-2ah"
-    if experiment_type == "REPRODUCABILITY CHECK 2":
+    if experiment_type == "Reproducability check (1)": 
+        model_repo = f"polejowska/detr-r50-cd45rb-all-2ah"
+    if experiment_type == "Reproducability check (2)":
         model_repo = f"polejowska/detr-r50-cd45rb-all-4ah"
-    if experiment_type == "REPRODUCABILITY CHECK 3":
+    if experiment_type == "Reproducability check (3)":
         model_repo = f"polejowska/detr-r50-cd45rb-all-8ah"
-    if experiment_type == "REPRODUCABILITY CHECK 4":
+    if experiment_type == "Reproducability check (4):
         model_repo = f"polejowska/detr-r50-cd45rb-all-16ah"
     model = DetrForObjectDetection.from_pretrained(model_repo)
 
@@ -99,15 +101,15 @@ with gr.Blocks(css=css) as app:
                 with gr.Column():
                     with gr.Row():
                         experiment_type = gr.Dropdown(
-                            value="DEFAULT",
+                            value="Parameters verification",
                             choices=[
-                                "REPRODUCABILITY CHECK 1",
-                                "REPRODUCABILITY CHECK 2",
-                                "REPRODUCABILITY CHECK 3",
-                                "REPRODUCABILITY CHECK 4",
-                                
+                                "Parameters verification",
+                                "Reproducability check (1)",
+                                "Reproducability check (2)",
+                                "Reproducability check (3)",
+                                "Reproducability check (3)",
                             ],
-                            label="Select an object detection model",
+                            label="Select an experiment type",
                             show_label=True,
                         )
                     with gr.Row():
@@ -148,7 +150,6 @@ with gr.Blocks(css=css) as app:
                             choices=[
                                 "ReLU",
                                 "GeLU",
-                                
                             ],
                             label="Select an activation function",
                             show_label=True,
@@ -166,7 +167,6 @@ with gr.Blocks(css=css) as app:
                             choices=[
                                 "True",
                                 "False",
-                                
                             ],
                             label="Use dilation",
                             show_label=True,