Update app.py

app.py

@@ -43,21 +43,38 @@ def get_model_config(hf_model: str) -> dict[str, Any]:
                 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
-            )
-        
+            if param_props_idx != -1:
+                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
+                )
+            else:
+                param_props_idx = model_page.find('data-target="ModelHeader"')
+                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;", '"')
+                    )["model"]["safetensors"]["total"]
+                    * 2
+                )
+
 
     # assume fp16 weights
     config["parameters"] = model_size / 2