AshBlanc commited on
Commit
2d1d65e
·
verified ·
1 Parent(s): 6878a82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,6 +5,8 @@ from transformers import pipeline
5
 
6
  # Load the gated model
7
  model_name = "RickyDeSkywalker/TheoremLlama"
 
 
8
  generator = pipeline('text-generation', model=model_name, token=os.environ.get("HF_TOKEN"))
9
 
10
  # Streamlit app
@@ -24,4 +26,4 @@ if st.button("Generate Lean 4 Code"):
24
  # Display the generated code
25
  st.code(generated_code, language='lean')
26
 
27
- st.markdown("Developed by [Your Name]")
 
5
 
6
  # Load the gated model
7
  model_name = "RickyDeSkywalker/TheoremLlama"
8
+ st.write(os.environ.get("HF_TOKEN"))
9
+ """
10
  generator = pipeline('text-generation', model=model_name, token=os.environ.get("HF_TOKEN"))
11
 
12
  # Streamlit app
 
26
  # Display the generated code
27
  st.code(generated_code, language='lean')
28
 
29
+ st.markdown("Developed by [Your Name]")"""