Update app.py

app.py

@@ -3,13 +3,15 @@ import torch
 import spaces
 from transformers import AutoModel
 
+hf_token = os.getenv("HF_TOKEN")
+
 @spaces.GPU
 def get_model_summary(model_name):
     # Check if CUDA is available and set the device accordingly
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
     # Load the model and move it to the selected device
-    model = AutoModel.from_pretrained(model_name).to(device)
+    model = AutoModel.from_pretrained(model_name, hf_token=hf_token).to(device)
     
     # Return the model's architecture as a string
     return str(model)