Update app.py

app.py

@@ -2,10 +2,11 @@ import gradio as gr
 from transformers import pipeline, AutoTokenizer
 import torch
 import spaces
+from huggingface_hub import secrets
 
 @spaces.GPU
 def load_model(model_name):
-    return pipeline("text-generation", model=model_name, device="cuda", torch_dtype=torch.float16, trust_remote_code=True)
+    return pipeline("text-generation", model=model_name, device="cuda", torch_dtype=torch.float16, trust_remote_code=True, token=secrets.get("token"))
 @spaces.GPU
 def generate(
     model_name,