Update app.py

app.py

@@ -3,13 +3,9 @@ import streamlit as st
 from huggingface_hub import login
 from transformers import pipeline
 
-# Authenticate with Hugging Face Hub
-HF_TOKEN = os.environ['HF_TOKEN']
-login(HF_TOKEN)
-
 # Load the gated model
 model_name = "RickyDeSkywalker/TheoremLlama"
-generator = pipeline('text-generation', model=model_name)
+generator = pipeline('text-generation', model=model_name, token=os.getenv("HF_TOKEN"))
 
 # Streamlit app
 st.title("Lean 4 Code Generation with TheoremLlama")