Update app.py
Browse files
app.py
CHANGED
|
@@ -86,7 +86,7 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
|
|
| 86 |
|
| 87 |
if selected_file is not None:
|
| 88 |
print(selected_file.rfilename)
|
| 89 |
-
component_wise_memory[component] =
|
| 90 |
|
| 91 |
return format_output(component_wise_memory)
|
| 92 |
|
|
|
|
| 86 |
|
| 87 |
if selected_file is not None:
|
| 88 |
print(selected_file.rfilename)
|
| 89 |
+
component_wise_memory[component] = selected_file.size
|
| 90 |
|
| 91 |
return format_output(component_wise_memory)
|
| 92 |
|