fixing ui

app.py

@@ -46,7 +46,9 @@ TASK_CATEGORIES = {
     "inpaint": {"color": "#ec4899", "category": "Multimodal"},
     "zero-shot-image-classification": {"color": "#ec4899", "category": "Multimodal"},
     "sentence-similarity": {"color": "#06b6d4", "category": "Similarity"},
-    "Auto": {"color": "#6b7280", "category": "Auto"},
+}
+
+TAGS = {
     "Feature Extraction": {"color": "#3b82f6", "category": "Feature Extraction"},
     "NLP": {"color": "#8b5cf6", "category": "NLP"},
     "Text Generation": {"color": "#10b981", "category": "Text Generation"},
@@ -64,6 +66,9 @@ def create_task_tag(task: str) -> str:
     if task in TASK_CATEGORIES:
         color = TASK_CATEGORIES[task]["color"]
         return f'<span style="background-color: {color}; color: white; padding: 2px 6px; border-radius: 12px; font-size: 0.75rem; font-weight: 500; margin: 1px;">{task}</span>'
+    elif task in TAGS:
+        color = TAGS[task]["color"]
+        return f'<span style="background-color: {color}; color: white; padding: 2px 6px; border-radius: 12px; font-size: 0.75rem; font-weight: 500; margin: 1px;">{task}</span>'
     else:
         return f'<span style="background-color: #6b7280; color: white; padding: 2px 6px; border-radius: 12px; font-size: 0.75rem; font-weight: 500; margin: 1px;">{task}</span>'
 
@@ -230,8 +235,7 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
                 input_task = gr.Dropdown(
                     choices=ALL_TASKS,
                     value="auto",
-                    label='Task',
-                    info='Select the task type for your model (auto will infer automatically)',
+                    label='Task (auto could infer task from model)',
                 )
             
             # Export button below the inputs
@@ -251,7 +255,6 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
             <div style="margin-bottom: 20px;">
                 <h3>🎨 Task Categories Legend</h3>
                 <div class="task-tags">
-                    {create_task_tag("Auto")} 
                     {create_task_tag("Feature Extraction")} 
                     {create_task_tag("NLP")}
                     {create_task_tag("Text Generation")}