Update app.py

app.py

@@ -1,6 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 import torch
+import os
+from huggingface_hub import login
+
+hf_token = os.environ["HF_TOKEN"]
+login(token=hf_token)
+
 
 device = 0 if torch.cuda.is_available() else -1
 pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2", device=device)