Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Some more fixes
Browse files- src/app.py +9 -8
src/app.py
CHANGED
@@ -33,7 +33,7 @@ def get_results(model_name: str, library: str, options: list, access_token: str)
|
|
33 |
)
|
34 |
for i, dtype in enumerate(options):
|
35 |
if stages["model"][i] != -1:
|
36 |
-
memory_values.loc[len(memory_values)] = [
|
37 |
dtype,
|
38 |
convert_bytes(stages["model"][i]),
|
39 |
convert_bytes(stages["gradients"][i]),
|
@@ -47,13 +47,14 @@ def get_results(model_name: str, library: str, options: list, access_token: str)
|
|
47 |
gr.update(visible=True, value=memory_values),
|
48 |
gr.update(visible=not has_discussion),
|
49 |
]
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
|
|
57 |
|
58 |
|
59 |
with gr.Blocks() as demo:
|
|
|
33 |
)
|
34 |
for i, dtype in enumerate(options):
|
35 |
if stages["model"][i] != -1:
|
36 |
+
memory_values.loc[len(memory_values.index)] = [
|
37 |
dtype,
|
38 |
convert_bytes(stages["model"][i]),
|
39 |
convert_bytes(stages["gradients"][i]),
|
|
|
47 |
gr.update(visible=True, value=memory_values),
|
48 |
gr.update(visible=not has_discussion),
|
49 |
]
|
50 |
+
else:
|
51 |
+
return [
|
52 |
+
title,
|
53 |
+
gr.update(visible=True, value=pd.DataFrame(data)),
|
54 |
+
gr.update(visible=False, value=""),
|
55 |
+
gr.update(visible=False, value=pd.DataFrame()),
|
56 |
+
gr.update(visible=not has_discussion),
|
57 |
+
]
|
58 |
|
59 |
|
60 |
with gr.Blocks() as demo:
|