Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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]")"""
|