Update app.py

app.py

@@ -10,7 +10,7 @@ model_name = "internlm/internlm2-math-plus-7b"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 #login(HF_TOKEN)
 
-generator = pipeline('text-generation', model=model_name, token=HF_TOKEN)
+generator = pipeline('text-generation', model=model_name, trust_remote_code=True, token=HF_TOKEN)
 
 # Function for generating Lean 4 code
 def generate_lean4_code(prompt):