Update app.py

app.py

@@ -1,7 +1,18 @@
+import os
 import gradio as gr
 from transformers import pipeline
+from huggingface_hub import login
 
-# Initialize the text generation pipeline with the Gemma 2-2B IT model
+# Read the token from the environment variable
+HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+
+# Authenticate with Hugging Face
+if HUGGINGFACE_TOKEN:
+    login(token=HUGGINGFACE_TOKEN)
+else:
+    raise ValueError("Hugging Face token not found in environment variables.")
+
+# Initialize the text generation pipeline
 pipe = pipeline("text-generation", model="google/gemma-2-2b-it")
 
 def generate_response(user_input):