pcuenq HF staff commited on
Commit
6e6214b
·
1 Parent(s): 0d00a12

Map internal task names to labels.

Browse files

They are consistent with Hub Tasks when possible.

Files changed (1) hide show
  1. app.py +23 -0
app.py CHANGED
@@ -48,6 +48,27 @@ push_mapping = {
48
  }
49
  push_labels = list(push_mapping.keys())
50
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  def error_str(error, title="Error", model=None, task=None, framework=None, compute_units=None, precision=None, tolerance=None, destination=None, open_discussion=True):
52
  if not error: return ""
53
 
@@ -123,6 +144,7 @@ def on_model_change(model):
123
 
124
  features = FeaturesManager.get_supported_features_for_model_type(model_type)
125
  tasks = list(features.keys())
 
126
 
127
  frameworks = supported_frameworks(model)
128
  selected_framework = frameworks[0] if len(frameworks) > 0 else None
@@ -194,6 +216,7 @@ def convert(model_id, task,
194
  push_destination, destination_model, token,
195
  progress=gr.Progress()):
196
  model_id = url_to_model_id(model_id)
 
197
  compute_units = compute_units_mapping[compute_units]
198
  precision = precision_mapping[precision]
199
  tolerance = tolerance_mapping[tolerance]
 
48
  }
49
  push_labels = list(push_mapping.keys())
50
 
51
+ tasks_mapping = {
52
+ "default": "Feature Extraction",
53
+ "causal-lm": "Text Generation",
54
+ "ctc": "CTC (Connectionist Temporal Classification)",
55
+ "image-classification": "Image Classification",
56
+ "image-segmentation": "Image Segmentation",
57
+ "masked-im": "Image Fill-Mask",
58
+ "masked-lm": "Fill-Mask",
59
+ "multiple-choice": "Multiple Choice",
60
+ "next-sentence-prediction": "Next Sentence Prediction",
61
+ "object-detection": "Object Detection",
62
+ "question-answering": "Question Answering",
63
+ "semantic-segmentation": "Semantic Segmentation",
64
+ "seq2seq-lm": "Text to Text Generation",
65
+ "sequence-classification": "Text Classification",
66
+ "speech-seq2seq": "Speech to Speech Generation",
67
+ "token-classification": "Token Classification",
68
+ }
69
+ reverse_tasks_mapping = {v: k for k, v in tasks_mapping.items()}
70
+ tasks_labels = list(tasks_mapping.keys())
71
+
72
  def error_str(error, title="Error", model=None, task=None, framework=None, compute_units=None, precision=None, tolerance=None, destination=None, open_discussion=True):
73
  if not error: return ""
74
 
 
144
 
145
  features = FeaturesManager.get_supported_features_for_model_type(model_type)
146
  tasks = list(features.keys())
147
+ tasks = [tasks_mapping[task] for task in tasks]
148
 
149
  frameworks = supported_frameworks(model)
150
  selected_framework = frameworks[0] if len(frameworks) > 0 else None
 
216
  push_destination, destination_model, token,
217
  progress=gr.Progress()):
218
  model_id = url_to_model_id(model_id)
219
+ task = reverse_tasks_mapping[task]
220
  compute_units = compute_units_mapping[compute_units]
221
  precision = precision_mapping[precision]
222
  tolerance = tolerance_mapping[tolerance]