Update app.py

app.py

@@ -3,6 +3,36 @@ import json
 import requests
 import pandas as pd  
 
+def update_task_options(framework):
+    config = {
+        "Custom":["Custom"],
+        "Diffusers":["Text To Image"],
+        "Transformers":[
+            "Text Classification", 
+            "Zero Shot Classification", 
+            "Token Classifiation", 
+            "Question Answering", 
+            "Fill Mask", 
+            "Summarization", 
+            "Translation", 
+            "Text to Text Generation", 
+            "Text Generation", 
+            "Feature Extraction", 
+            "Image Classification", 
+            "Automatic Speech Recognition", 
+            "Audio Classification", 
+            "Object Detection", 
+            "Image Segmentation", 
+            "Table Question Answering", 
+            "Conversational", 
+            "Visual Question Answering", 
+            "Zero Shot Image Classification"]
+    }
+    return gr.Dropdown.update(
+        choices=config[framework],
+        value=config[framework][0] if len(config[framework]) > 0 else None
+    )
+
 def update_regions(provider):
     available_regions = []
     
@@ -297,13 +327,15 @@ with gr.Blocks() as interface:
             """)      
         
         with gr.Row():
-            task_selector = gr.Textbox(
+            framework_selector = gr.Dropdown(
+                choices = ["Custom", "Diffusers", "Transformers"],
                 value="Custom",
                 interactive=True,
                 show_label=False,
             )
 
-            framework_selector = gr.Textbox(
+            task_selector = gr.Dropdown(
+                [],
                 value="Custom",
                 interactive=True,
                 show_label=False,
@@ -507,6 +539,7 @@ with gr.Blocks() as interface:
         )
 
     # Deploy Tab Events
+    framework_selector.change(update_task_options, inputs=framework_selector, outputs=task_selector)
     provider_selector.change(update_regions, inputs=provider_selector, outputs=region_selector)
     region_selector.change(update_compute_options, inputs=[provider_selector, region_selector], outputs=compute_selector)
     submit_button.click(