Spaces:
Running
Running
Update app.py
Browse filesadding existing models
app.py
CHANGED
@@ -120,7 +120,7 @@ def add_new_eval(
|
|
120 |
"precision": precision,
|
121 |
"status": "PENDING",
|
122 |
"submitted_time": current_time,
|
123 |
-
"task": task
|
124 |
"likes": model_info.likes,
|
125 |
"params": model_size}
|
126 |
#"license": license,
|
@@ -152,6 +152,24 @@ def add_new_eval(
|
|
152 |
print("Starting compute space at %s " % COMPUTE_SPACE)
|
153 |
return start_compute_space()
|
154 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
155 |
|
156 |
|
157 |
|
@@ -191,5 +209,6 @@ with gr.Blocks() as demo:
|
|
191 |
],
|
192 |
outputs=submission_result,
|
193 |
)
|
194 |
-
|
|
|
195 |
demo.launch()
|
|
|
120 |
"precision": precision,
|
121 |
"status": "PENDING",
|
122 |
"submitted_time": current_time,
|
123 |
+
"task": task,
|
124 |
"likes": model_info.likes,
|
125 |
"params": model_size}
|
126 |
#"license": license,
|
|
|
152 |
print("Starting compute space at %s " % COMPUTE_SPACE)
|
153 |
return start_compute_space()
|
154 |
|
155 |
+
def print_existing_models():
|
156 |
+
model_dict= []
|
157 |
+
requests_dataset = snapshot_download('EnergyStarAI/requests_debug', token=TOKEN, repo_type="dataset", allow_patterns="*.json")
|
158 |
+
for dir, path, files in os.walk(requests_dataset):
|
159 |
+
for fid in files:
|
160 |
+
file_path = os.path.join(dir, fid)
|
161 |
+
with open(file_path) as fp:
|
162 |
+
request = json.load(fp)
|
163 |
+
status = request['status']
|
164 |
+
if status == 'COMPLETE':
|
165 |
+
status = request['status']
|
166 |
+
model = request['model']
|
167 |
+
task = request['task']
|
168 |
+
model_dict.append([model, task])
|
169 |
+
return model_dict
|
170 |
+
|
171 |
+
|
172 |
+
|
173 |
|
174 |
|
175 |
|
|
|
209 |
],
|
210 |
outputs=submission_result,
|
211 |
)
|
212 |
+
with gr.Row():
|
213 |
+
gr.Dataframe(print_existing_models())
|
214 |
demo.launch()
|