removed suggestions

app.py

@@ -345,33 +345,15 @@ def get_class_specific_attention(image, model_name, class_query):
 
 
 # ---------------------------
-# Get Class Suggestions
+# Sample Classes
 # ---------------------------
-def get_class_suggestions(query, model_name):
-    try:
-        if not query or len(query.strip()) < 2:
-            return gr.Dropdown(choices=[], value=None)
-        
-        query_lower = query.lower().strip()
-        suggestions = []
-        
-        model, extractor = load_model(model_name)
-        
-        if MODEL_CONFIGS[model_name]["type"] == "hf":
-            labels = list(model.config.id2label.values())
-        else:
-            labels = IMAGENET_LABELS
-        
-        for label in labels:
-            if query_lower in label.lower():
-                suggestions.append(label)
-                if len(suggestions) >= 10:  # Limit to 10 suggestions
-                    break
-        
-        return gr.Dropdown(choices=suggestions, value=None)
-    except Exception as e:
-        print(f"Error getting suggestions: {e}")
-        return gr.Dropdown(choices=[], value=None)
+SAMPLE_CLASSES = [
+    "cat", "dog", "tiger", "lion", "elephant",
+    "car", "truck", "airplane", "ship", "train",
+    "pizza", "hamburger", "coffee", "banana", "apple",
+    "chair", "table", "laptop", "keyboard", "mouse",
+    "person", "bicycle", "building", "tree", "flower"
+]
 
 
 # ---------------------------
@@ -416,13 +398,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             class_input = gr.Textbox(
                 label="🔍 Enter Class Name",
                 placeholder="e.g., cat, dog, car, pizza...",
-                info="Start typing to see suggestions"
+                info="Type any ImageNet class name"
             )
-            class_suggestions = gr.Dropdown(
-                label="💡 Suggestions (click to use)",
-                choices=[],
-                interactive=True,
-                allow_custom_value=False
+            gr.Markdown("**💡 Sample classes to try:**")
+            sample_buttons = gr.Radio(
+                choices=SAMPLE_CLASSES,
+                label="Click to auto-fill",
+                interactive=True
             )
             class_button = gr.Button("🎯 Generate Class-Specific Attention", variant="primary")
             
@@ -445,17 +427,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[output_label, output_image, processed_image]
     )
     
-    # Update suggestions as user types
-    class_input.change(
-        get_class_suggestions,
-        inputs=[class_input, model_dropdown],
-        outputs=[class_suggestions]
-    )
-    
-    # When user selects from suggestions, update the text input
-    class_suggestions.select(
+    # When user selects a sample class, update the text input
+    sample_buttons.change(
         lambda x: x,
-        inputs=[class_suggestions],
+        inputs=[sample_buttons],
         outputs=[class_input]
     )