Spaces:

AshBlanc
/

demo-prover

Running

AshBlanc commited on Nov 28, 2024

Commit

430bd31

verified ·

1 Parent(s): a0a60c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import streamlit as st
 from transformers import pipeline
 # Load the gated model
-model_name = "RickyDeSkywalker/TheoremLlama"
 token = os.environ.get("HF_TOKEN")
 generator = pipeline('text-generation', model=model_name, token=token)

 from transformers import pipeline
 # Load the gated model
+model_name = "internlm/internlm2-math-plus-7b"
 token = os.environ.get("HF_TOKEN")
 generator = pipeline('text-generation', model=model_name, token=token)