fix: hf adjustments

app.py

@@ -3,41 +3,69 @@ from safetensors.torch import load_file
 from model_loader import get_top_layers, load_model_summary, load_config
 import tempfile
 import os
+import requests
+import json
 
-def inspect_model(model_file, config_file=None):
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".safetensors") as tmp:
-        tmp.write(model_file.read())
-        model_path = tmp.name
+def inspect_model(model_id, config_file=None):
+    if not model_id or '/' not in model_id:
+        return "Please provide a valid model ID in the format username/modelname", "No config loaded."
+    
+    username, modelname = model_id.split('/', 1)
+    
+    model_url = f"https://huggingface.co/{username}/{modelname}/resolve/main/model.safetensors"
+    
+    try:
+        response = requests.get(model_url)
+        response.raise_for_status()  
+        
 
-    summary = load_model_summary(model_path)
-    state_dict = load_file(model_path)
-    top_layers = get_top_layers(state_dict, summary["total_params"])
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".safetensors") as tmp:
+            tmp.write(response.content)
+            model_path = tmp.name
+        
 
-    top_layers_str = "\n".join([
-        f"{layer['name']}: shape={layer['shape']}, params={layer['params']:,} ({layer['percent']}%)"
-        for layer in top_layers
-    ])
+        summary = load_model_summary(model_path)
+        state_dict = load_file(model_path)
+        top_layers = get_top_layers(state_dict, summary["total_params"])
+        top_layers_str = "\n".join([
+            f"{layer['name']}: shape={layer['shape']}, params={layer['params']:,} ({layer['percent']}%)"
+            for layer in top_layers
+        ])
+        
 
-
-    config_data = {}
-    if config_file is not None:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".json") as tmp_cfg:
-            tmp_cfg.write(config_file.read())
-            config_data = load_config(tmp_cfg.name)
-
-    config_str = "\n".join([f"{k}: {v}" for k, v in config_data.items()]) if config_data else "No config loaded."
-
-    return (
-        f" Total tensors: {summary['num_tensors']}\n"
-        f" Total parameters: {summary['total_params']:,}\n\n"
-        f" Top Layers:\n{top_layers_str}",
-        config_str
-    )
+        config_data = {}
+        if config_file is not None:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".json") as tmp_cfg:
+                tmp_cfg.write(config_file.read())
+                config_data = load_config(tmp_cfg.name)
+                os.unlink(tmp_cfg.name)  
+        else:
+            config_url = f"https://huggingface.co/{username}/{modelname}/resolve/main/config.json"
+            try:
+                config_response = requests.get(config_url)
+                config_response.raise_for_status()
+                config_data = json.loads(config_response.content)
+            except Exception as e:
+                pass  
+        
+        config_str = "\n".join([f"{k}: {v}" for k, v in config_data.items()]) if config_data else "No config loaded."
+        
+        os.unlink(model_path)
+        
+        return (
+            f" Total tensors: {summary['num_tensors']}\n"
+            f" Total parameters: {summary['total_params']:,}\n\n"
+            f" Top Layers:\n{top_layers_str}",
+            config_str
+        )
+    
+    except Exception as e:
+        return f"Error: {str(e)}", "No config loaded."
 
 gr.Interface(
     fn=inspect_model,
     inputs=[
-        gr.File(label="Upload model (.safetensors)", type="binary"),
+        gr.Textbox(label="Model ID from HuggingFace", placeholder="username/modelname", lines=1),
         gr.File(label="Upload config.json (optional)", type="binary")
     ],
     outputs=[
@@ -45,5 +73,5 @@ gr.Interface(
         gr.Textbox(label="Config", lines=10)
     ],
     title="Model Inspector",
-    description="Upload a .safetensors file to analyze its structure, parameter count, and optionally its config.json"
-).launch()
+    description="Enter a HuggingFace model ID in the format username/modelname to analyze its structure, parameter count, and configuration."
+).launch()