pcuenq HF staff commited on
Commit
4587527
·
1 Parent(s): deafe70

Ignore `with-past` for now.

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -167,14 +167,17 @@ def on_model_change(model):
167
  config = json.loads(config_json)
168
  model_type = config["model_type"]
169
 
 
170
  features = FeaturesManager.get_supported_features_for_model_type(model_type)
171
  tasks = list(features.keys())
 
172
 
173
  model_info = retrieve_model_info(model)
174
  frameworks = model_info["frameworks"]
175
  selected_framework = frameworks[0] if len(frameworks) > 0 else None
176
 
177
  pipeline_tag = model_info["pipeline_tag"]
 
178
  # Select the task corresponding to the pipeline tag
179
  if tasks:
180
  if pipeline_tag in tags_to_tasks_mapping:
 
167
  config = json.loads(config_json)
168
  model_type = config["model_type"]
169
 
170
+ # Ignore `-with-past` for now
171
  features = FeaturesManager.get_supported_features_for_model_type(model_type)
172
  tasks = list(features.keys())
173
+ tasks = [task for task in tasks if "-with-past" not in task]
174
 
175
  model_info = retrieve_model_info(model)
176
  frameworks = model_info["frameworks"]
177
  selected_framework = frameworks[0] if len(frameworks) > 0 else None
178
 
179
  pipeline_tag = model_info["pipeline_tag"]
180
+ # print(pipeline_tag)
181
  # Select the task corresponding to the pipeline tag
182
  if tasks:
183
  if pipeline_tag in tags_to_tasks_mapping: