Update app.py

app.py

@@ -2,14 +2,15 @@ import gradio as gr
 import torch
 import torchvision
 from transformers import pipeline
+auth_token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
 
 # Initialize the pipeline
 pipe = pipeline(
     "text-generation",
     model="DataIntelligenceTeam/mistral_7B_NER",
     torch_dtype=torch.bfloat16,
-    device_map="auto"
-)
+    device_map="auto",
+    use_auth_token=auth_token)
 
 # Function to generate output based on input
 def generate_output(input_text):