Update app.py

changing list to dict for tasks

app.py

@@ -18,8 +18,10 @@ API = HfApi(token=TOKEN)
 
 
 
-tasks = ['ASR', 'Object Detection', 'Text Classification', 'Image Captioning', 'Question Answering', 'Text Generation', 'Image Classification',
-        'Sentence Similarity', 'Image Generation', 'Summarization']
+task_mappings = {'ASR':'text-to-speech', 'Object Detection': 'object-detection', 'Text Classification': 'text-classification', 
+                 'Image Captioning':'image-to-text', 'Question Answering':'question-answering', 'Text Generation': 'text-generation',
+                 'Image Classification':'image-classification', 'Sentence Similarity': 'sentence-similarity',
+                 'Image Generation':'image-generation', 'Summarization':'summarization'}
 
 @dataclass
 class ModelDetails:
@@ -52,7 +54,7 @@ def add_new_eval(
     requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
     requests_dset = requests.to_pandas()
     model_list= requests_dset[requests_dset['status'] == 'COMPLETED']['model'].tolist()
-    task_models = list(API.list_models(filter=task.lower().replace(' ','-')))
+    task_models = list(API.list_models(filter=task_mappings[task]))
     task_model_names = [m.id for m in task_models]
     if repo_id in model_list:
         return 'This model has already been run!'
@@ -116,7 +118,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             task = gr.Dropdown(
-                choices=tasks,
+                choices=task_mappings.keys(),
                 label="Choose a benchmark task",
                 value = 'Text Generation',
                 multiselect=False,