Update app.py

app.py

@@ -5,9 +5,10 @@ from transformers import pipeline
 
 # Load the gated model
 model_name = "RickyDeSkywalker/TheoremLlama"
-st.write(os.environ.get("HF_TOKEN"))
-"""
-generator = pipeline('text-generation', model=model_name, token=os.environ.get("HF_TOKEN"))
+HF_TOKEN = os.environ.get("HF_TOKEN")
+login(HF_TOKEN)
+
+generator = pipeline('text-generation', model=model_name, token=HF_TOKEN)
 
 # Streamlit app
 st.title("Lean 4 Code Generation with TheoremLlama")
@@ -26,4 +27,4 @@ if st.button("Generate Lean 4 Code"):
     # Display the generated code
     st.code(generated_code, language='lean')
 
-st.markdown("Developed by [Your Name]")"""
+st.markdown("Developed by [Your Name]")