sasha HF Staff commited on
Commit
48774fd
·
verified ·
1 Parent(s): 2660d71

Update app.py

Browse files

changing `model_type` to `task`

Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -16,6 +16,9 @@ REQUESTS_DATASET_PATH = f"{OWNER}/requests_debug"
16
  TOKEN = os.environ.get("DEBUG")
17
  API = HfApi(token=TOKEN)
18
 
 
 
 
19
 
20
 
21
  ##### Data classes need for the leaderboard Submit Model menu. #####
@@ -95,7 +98,7 @@ def add_new_eval(
95
  revision: str,
96
  precision: str,
97
  weight_type: str,
98
- model_type: str,
99
  ):
100
  model_owner = repo_id.split("/")[0]
101
  model_name = repo_id.split("/")[1]
@@ -103,8 +106,6 @@ def add_new_eval(
103
 
104
  current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
105
 
106
- #if model_type is None or model_type == "":
107
- # return styled_error("Please select a model type.")
108
 
109
  # Does the model actually exist?
110
  #if revision == "":
@@ -151,7 +152,7 @@ def add_new_eval(
151
  "weight_type": weight_type,
152
  "status": "PENDING",
153
  "submitted_time": current_time,
154
- "model_type": model_type,
155
  "likes": model_info.likes,
156
  "params": model_size}
157
  #"license": license,
@@ -201,11 +202,10 @@ with gr.Blocks() as demo:
201
  with gr.Column():
202
  model_name_textbox = gr.Textbox(label="Model name", value="lvwerra/distilbert-imdb")
203
  revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
204
- model_type = gr.Dropdown(
205
- choices=[t.to_str(" : ") for t in ModelType if t != ModelType.Unknown],
206
- label="Model type",
207
  multiselect=False,
208
- value=ModelType.PT.to_str(" : "),
209
  interactive=True,
210
  )
211
 
@@ -236,7 +236,7 @@ with gr.Blocks() as demo:
236
  revision_name_textbox,
237
  precision,
238
  weight_type,
239
- model_type,
240
  ],
241
  outputs=submission_result,
242
  )
 
16
  TOKEN = os.environ.get("DEBUG")
17
  API = HfApi(token=TOKEN)
18
 
19
+ tasks = ['asr.csv', 'object_detection.csv', 'text_classification.csv', 'image_captioning.csv',
20
+ 'question_answering.csv', 'text_generation.csv', 'image_classification.csv',
21
+ 'sentence_similarity.csv', 'image_generation.csv', 'summarization.csv']
22
 
23
 
24
  ##### Data classes need for the leaderboard Submit Model menu. #####
 
98
  revision: str,
99
  precision: str,
100
  weight_type: str,
101
+ task: str,
102
  ):
103
  model_owner = repo_id.split("/")[0]
104
  model_name = repo_id.split("/")[1]
 
106
 
107
  current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
108
 
 
 
109
 
110
  # Does the model actually exist?
111
  #if revision == "":
 
152
  "weight_type": weight_type,
153
  "status": "PENDING",
154
  "submitted_time": current_time,
155
+ "task": task,
156
  "likes": model_info.likes,
157
  "params": model_size}
158
  #"license": license,
 
202
  with gr.Column():
203
  model_name_textbox = gr.Textbox(label="Model name", value="lvwerra/distilbert-imdb")
204
  revision_name_textbox = gr.Textbox(label="Revision commit", placeholder="main")
205
+ task = gr.Dropdown(
206
+ choices=tasks,
207
+ label="Tasks",
208
  multiselect=False,
 
209
  interactive=True,
210
  )
211
 
 
236
  revision_name_textbox,
237
  precision,
238
  weight_type,
239
+ task,
240
  ],
241
  outputs=submission_result,
242
  )