UPDATED ERRORS

app.py

@@ -10,7 +10,6 @@ import torchvision.transforms as T
 import urllib.request
 import json
 import cv2
-import uuid
 
 # Model Configs
 MODEL_CONFIGS = {
@@ -237,6 +236,9 @@ def predict(image, model_name, noise_level):
         if model_name is None:
             return {"Error": "Please select a model"}, None, None
 
+        # Extract model name from dropdown (remove description)
+        model_name = model_name.split(" - ")[0]
+
         if noise_level > 0:
             image = add_adversarial_noise(image, noise_level)
 
@@ -287,6 +289,9 @@ def get_class_specific_attention(image, model_name, class_query):
         if not class_query or class_query.strip() == "":
             return None, None, "Please enter a class name"
 
+        # Extract model name from dropdown (remove description)
+        model_name = model_name.split(" - ")[0]
+
         class_query_lower = class_query.lower().strip()
         matching_idx = None
         matched_label = None
@@ -355,7 +360,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="gray",
             with gr.Row(variant="panel"):
                 with gr.Column(scale=1, min_width=300):
                     gr.Markdown("### 📷 Input")
-                    input_image = gr.Image(type="pil", label="Upload Image", height=300, interactive=True, tool="editor")
+                    input_image = gr.Image(type="pil", label="Upload Image", height=300, interactive=True)
                     model_dropdown = gr.Dropdown(
                         choices=[f"{name} - {MODEL_CONFIGS[name]['desc']}" for name in MODEL_CONFIGS.keys()],
                         label="Select Model",
@@ -371,7 +376,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="gray",
                             info="Add noise to test model robustness",
                             interactive=True
                         )
-                    run_button = gr.Button("🚀 Run Prediction", variant="primary", scale=0)
+                    run_button = gr.Button("🚀 Run Prediction", variant="primary")
                 
                 with gr.Column(scale=2):
                     gr.Markdown("### 📊 Results")