Update app.py

app.py

@@ -76,9 +76,17 @@ else:
 
 import gradio as gr
 from transformers import pipeline
+import os
+from huggingface_hub import login
+
+# Authenticate with HF Hub
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
 
 # Load a small, efficient model
-generator = pipeline("text-generation", model="distilgpt2")
+
+generator = pipeline("text-generation", model="distilgpt2", token=hf_token)
 
 def generate_text(prompt):
     result = generator(prompt, max_length=50, num_return_sequences=1)
@@ -95,3 +103,5 @@ demo = gr.Interface(
 # Keep the app alive on Hugging Face Spaces
 demo.launch(server_name="0.0.0.0", server_port=7860)
 
+
+