sayakpaul HF staff commited on
Commit
59db7fd
1 Parent(s): 1ea8dd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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] = bytes_to_giga_bytes(selected_file.size)
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