fix

app.py

@@ -73,13 +73,25 @@ def load_model_index(pipeline_id, token=None, revision=None):
 
 
 def get_individual_model_memory(id, token, variant, extension):
+    # Retrieve all files in the repository.
     files_in_repo = model_info(id, token=token, files_metadata=True).siblings
-    candidates = [x for x in files_in_repo if extension in x.rfilename]
+
+    # Filter files by extension and variant (if provided).
     if variant:
-        candidate = list(filter(lambda x: variant in x.rfilename, candidates))[0]
+        candidates = [x for x in files_in_repo if (extension in x.rfilename) and (variant in x.rfilename)]
+        if not candidates:
+            raise ValueError(f"Requested variant ({variant}) for {id} couldn't be found with {extension} extension.")
     else:
-        candidate = list(filter(lambda x: all(var not in x.rfilename for var in ALLOWED_VARIANTS[1:]), candidates))[0]
-    return candidate.size
+        candidates = [
+            x
+            for x in files_in_repo
+            if (extension in x.rfilename) and all(var not in x.rfilename for var in ALLOWED_VARIANTS[1:])
+        ]
+        if not candidates:
+            raise ValueError(f"No file for {id} could be found with {extension} extension without specified variants.")
+
+    # Return the size of the first matching file.
+    return candidates[0].size
 
 
 def get_component_wise_memory(