Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import pandas as pd
 import json
 import requests
+from html.parser import HTMLParser
 
 quants = {
     "Q2_K": 3.35,
@@ -19,6 +20,17 @@ quants = {
     "Q8_0": 8.5,
 }
 
+class SvelteHydratorExtractor(HTMLParser):
+    def __init__(self):
+        self.data = None
+        super().__init__()
+
+    def handle_starttag(self, tag, attrs):
+        print("Start tag:", tag)
+        for attr in attrs:
+            if attr[0] == "data-props":
+                self.data = attr[1].replace("&quot:", '"')
+
 
 def calc_model_size(parameters: int, quant: float) -> int:
     return parameters * quant // 8
@@ -45,37 +57,28 @@ def get_model_config(hf_model: str) -> dict[str, Any]:
             param_props_idx = model_page.find('data-target="ModelSafetensorsParams"')
             if param_props_idx != -1:
                 param_props_start = model_page.rfind("<div", 0, param_props_idx)
+                param_props_end = model_page.find(">", param_props_idx)
+                extractor = SvelteHydratorExtractor()
+                extractor.feed(model_page[param_props_start:param_props_end + 1])
                 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;", '"')
+                        extractor.data
                     )["safetensors"]["total"]
                     * 2
                 )
             else:
                 param_props_idx = model_page.find('data-target="ModelHeader"')
                 param_props_start = model_page.rfind("<div", 0, param_props_idx)
+                param_props_end = model_page.find(">", param_props_idx)
+                extractor = SvelteHydratorExtractor()
+                extractor.feed(model_page[param_props_start:param_props_end + 1])
                 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;", '"')
+                        extractor.data
                     )["model"]["safetensors"]["total"]
                     * 2
                 )
 
-
     # assume fp16 weights
     config["parameters"] = model_size / 2
     return config