Update app.py

app.py

@@ -9,6 +9,7 @@ import onnx.numpy_helper
 from huggingface_hub import HfApi
 from tempfile import NamedTemporaryFile
 import time
+from huggingface_hub import HfApi, login
 
 def analyze_weight_data(tensor):
     """Analyze tensor data"""
@@ -150,7 +151,22 @@ def format_analysis_text(analysis):
     return text
 
 def save_to_hf_dataset(analysis, model_name):
-    """Save analysis results to Hugging Face dataset"""
+    """
+    Save analysis results to Hugging Face dataset using token from environment variable
+    
+    Args:
+        analysis: Analysis results to save
+        model_name: Name of the model being analyzed
+    """
+    # Get token from environment variable
+    hf_token = os.getenv('HF_TOKEN')
+    if not hf_token:
+        raise ValueError("HF_TOKEN environment variable not found")
+        
+    # Login with token
+    login(token=hf_token)
+    
+    # Initialize API
     api = HfApi()
     
     # Create temporary files
@@ -163,7 +179,7 @@ def save_to_hf_dataset(analysis, model_name):
     with NamedTemporaryFile(mode='w', suffix='.json', delete=False) as json_file:
         json.dump(analysis, json_file, indent=2)
         json_path = json_file.name
-    
+        
     try:
         # Upload files to dataset
         api.upload_file(
@@ -175,10 +191,11 @@ def save_to_hf_dataset(analysis, model_name):
         
         api.upload_file(
             path_or_fileobj=json_path,
-            path_in_repo=f"analysis_{model_name}_{timestamp}.json",
+            path_in_repo=f"analysis_{model_name}_{timestamp}.json", 
             repo_id="Arrcttacsrks/OnnxAnalysisData",
             repo_type="dataset"
         )
+        
     finally:
         # Clean up temporary files
         os.unlink(txt_path)