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 |
|