Fetch files metadata directly from model_info call

app.py

@@ -1,4 +1,4 @@
-from huggingface_hub import hf_hub_url, get_hf_file_metadata, model_info
+from huggingface_hub import model_info
 import gradio as gr
 
 
@@ -22,7 +22,7 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
     print(pipeline_id, variant, revision, extension)
     component_wise_memory = {}
 
-    files_in_repo = model_info(pipeline_id, revision=revision, token=token).siblings
+    files_in_repo = model_info(pipeline_id, revision=revision, token=token, files_metadata=True).siblings
 
     for current_file in files_in_repo:
         if all(
@@ -44,10 +44,7 @@ def get_component_wise_memory(pipeline_id, token=None, variant=None, revision=No
                         filename = current_file.rfilename
 
                     if filename is not None:
-                        hub_url = hf_hub_url(repo_id=pipeline_id, filename=filename)
-                        file_metadata = get_hf_file_metadata(hub_url)
-
-                    component_wise_memory[component] = bytes_to_giga_bytes(file_metadata.size)
+                        component_wise_memory[component] = bytes_to_giga_bytes(current_file.size)
 
     return component_wise_memory