Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 model_name = "internlm/internlm2-math-plus-7b"
 token = os.environ.get("HF_TOKEN")
 
-generator = pipeline('text-generation', model=model_name, token=token)
+generator = pipeline('text-generation', model=model_name, token=token, trust_remote_code=True)
 
 # Streamlit app
 st.title("Lean 4 Code Generation with TheoremLlama")