Files changed (1) hide show
  1. app.py +2 -1
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