Update app.py

app.py

@@ -1,6 +1,12 @@
+import os
 import streamlit as st
+from huggingface_hub import login
 from transformers import pipeline
 
+# Authenticate with Hugging Face Hub
+user_token = os.getenv(HF_TOKEN)  # Replace with your actual Hugging Face user token
+login(user_token)
+
 # Load the gated model
 model_name = "RickyDeSkywalker/TheoremLlama"
 generator = pipeline('text-generation', model=model_name)