Update app.py

app.py

@@ -2,6 +2,12 @@ import gradio as gr
 import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 import re
+import os
+from huggingface_hub import login
+
+# Authenticate with Hugging Face
+if "HF_TOKEN" in os.environ:
+    login(token=os.environ["HF_TOKEN"])
 
 # Global variables to store the models
 atlas_pipe = None