Hugging Face's logo

Spaces:
ahmedghani
/
Inference-Endpoint-Deployment
Runtime error

App Files Community
ahmedghani commited on
Commit
8e2e411
·
1 Parent(s): c95d2f7

bug fix

Browse files
Files changed (1) hide show
  1. app.py +17 -19
app.py CHANGED
@@ -8,25 +8,23 @@ def update_task_options(framework):
8
  "Custom":["Custom"],
9
  "Diffusers":["Text To Image"],
10
  "Transformers":[
11
- "Text Classification",
12
- "Zero Shot Classification",
13
- "Token Classifiation",
14
- "Question Answering",
15
- "Fill Mask",
16
- "Summarization",
17
- "Translation",
18
- "Text to Text Generation",
19
- "Text Generation",
20
- "Feature Extraction",
21
- "Image Classification",
22
- "Automatic Speech Recognition",
23
- "Audio Classification",
24
- "Object Detection",
25
- "Image Segmentation",
26
- "Table Question Answering",
27
- "Conversational",
28
- "Visual Question Answering",
29
- "Zero Shot Image Classification"]
30
  }
31
  return gr.Dropdown.update(
32
  choices=config[framework],
 
8
  "Custom":["Custom"],
9
  "Diffusers":["Text To Image"],
10
  "Transformers":[
11
+ "text-classification",
12
+ "zero-shot-classification",
13
+ "token-classification",
14
+ "question-answering",
15
+ "fill-mask",
16
+ "summarization",
17
+ "translation",
18
+ "text2text-generation",
19
+ "text-generation",
20
+ "feature-extraction",
21
+ "image-classification",
22
+ "automatic-speech-recognition",
23
+ "audio-classification",
24
+ "object-detection",
25
+ "image-segmentation",
26
+ "table-question-answering"
27
+ ]
 
 
28
  }
29
  return gr.Dropdown.update(
30
  choices=config[framework],