Add "model.safetensors.index.json" support

app.py

@@ -837,6 +837,17 @@ def get_dim_seq_size(model):
         url = hf_hub_url(model.modelId, filename="model.safetensors")
         meta = get_hf_file_metadata(url)
         size = round(meta.size / 1e9, 2)
+    elif "model.safetensors.index.json" in filenames:
+        index_path = hf_hub_download(model.modelId, filename="model.safetensors.index.json")
+        """
+        {
+        "metadata": {
+            "total_size": 14483464192
+        },....
+        """
+        size = json.load(open(index_path))
+        if ("metadata" in size) and ("total_size" in size["metadata"]):
+            size = round(size["metadata"]["total_size"] / 1e9, 2)
     return dim, seq, size
 
 def make_datasets_clickable(df):