Update app.py

app.py

@@ -27,18 +27,39 @@ def get_model_config(hf_model: str) -> dict[str, Any]:
     config = requests.get(
         f"https://huggingface.co/{hf_model}/raw/main/config.json"
     ).json()
-    model_index = {}
+    model_size = 0
     try:
-        model_index = requests.get(
+        model_size = requests.get(
             f"https://huggingface.co/{hf_model}/raw/main/model.safetensors.index.json"
-        ).json()
+        ).json()["metadta"]["total_size"]
     except:
-        model_index = requests.get(
-            f"https://huggingface.co/{hf_model}/raw/main/pytorch_model.bin.index.json"
-        ).json()
+        try:
+            model_size = requests.get(
+                f"https://huggingface.co/{hf_model}/raw/main/pytorch_model.bin.index.json"
+            ).json()["metadta"]["total_size"]
+        except:
+            model_page = requests.get(
+                f"https://huggingface.co/{hf_model}"
+            ).text
+            param_props_idx = model_page.find('data-target="ModelSafetensorsParams"')
+            param_props_start = model_page.rfind("<div", 0, param_props_idx)
+            model_size = (
+                json.loads(
+                    [
+                        prop
+                        for prop in model_page[param_props_start:param_props_idx].split(" ")
+                        if prop.startswith("data-props=")
+                    ][0]
+                    .split("=")[1]
+                    .replace('"', "")
+                    .replace("&quot;", '"')
+                )["safetensors"]["total"]
+                * 2
+            )
+        
 
     # assume fp16 weights
-    config["parameters"] = model_index["metadata"]["total_size"] / 2
+    config["parameters"] = model_size / 2
     return config