Update app.py

app.py

@@ -4,6 +4,10 @@ import random
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import spaces
+from huggingface_hub import login
+import os
+if "HF_API_TOKEN" in os.environ:
+    login(token=os.environ["HF_API_TOKEN"])
 
 device = "cuda" if torch.cuda.is_available() else "cpu"