Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import pipeline
 
 # Load the gated model
 model_name = "RickyDeSkywalker/TheoremLlama"
-generator = pipeline('text-generation', model=model_name, token=os.getenv("HF_TOKEN"))
+generator = pipeline('text-generation', model=model_name, token=os.environ.get("HF_TOKEN"))
 
 # Streamlit app
 st.title("Lean 4 Code Generation with TheoremLlama")