Update app.py

app.py

@@ -3,6 +3,28 @@ import torch
 from transformers import pipeline
 import gradio as gr
 
+import os
+
+
+def init_huggingface_auth():
+    """
+    Initialize Hugging Face authentication in a Space environment
+    """
+    token = os.environ.get('HUGGINGFACE_TOKEN')
+    if token:
+        try:
+            login(token=token)
+            print("Successfully logged in to Hugging Face!")
+        except Exception as e:
+            print(f"Error during login: {str(e)}")
+            return False
+    else:
+        print("No Hugging Face token found in environment variables")
+        return False
+    return True
+
+if not init_huggingface_auth():
+        print("Warning: Hugging Face authentication failed")
 # Initialize the pipeline
 pipe = pipeline(
     "text-generation",