Spaces:
Running
Running
Update app.py
#1
by
AshBlanc
- opened
app.py
CHANGED
@@ -6,7 +6,8 @@ from transformers import pipeline
|
|
6 |
# Load the gated model
|
7 |
#model_name = "RickyDeSkywalker/TheoremLlama"
|
8 |
#model_name = "unsloth/Llama-3.2-1B-Instruct"
|
9 |
-
model_name = "internlm/internlm2-math-plus-7b"
|
|
|
10 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
11 |
#login(HF_TOKEN)
|
12 |
|
|
|
6 |
# Load the gated model
|
7 |
#model_name = "RickyDeSkywalker/TheoremLlama"
|
8 |
#model_name = "unsloth/Llama-3.2-1B-Instruct"
|
9 |
+
#model_name = "internlm/internlm2-math-plus-7b"
|
10 |
+
model_name = "deepseek-ai/DeepSeek-Prover-V1.5-RL"
|
11 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
12 |
#login(HF_TOKEN)
|
13 |
|