Update app.py

app.py

@@ -3,9 +3,8 @@ from pydantic import BaseModel
 from fastapi.responses import JSONResponse
 from transformers import pipeline
 
-# Load an instruction-tuned model for generation
-# This model must be supported on CPU; you can change it if needed
-pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", max_new_tokens=100)
+# Load a small, CPU-friendly instruction-following model
+pipe = pipeline("text2text-generation", model="google/flan-t5-small", max_new_tokens=100)
 
 app = FastAPI()
 
@@ -14,10 +13,9 @@ class LaTeXRequest(BaseModel):
 
 @app.post("/fix")
 async def fix_latex(data: LaTeXRequest):
-    prompt = f"Fix this malformed LaTeX expression so it's valid and simple:\n```{data.latex}```\nFixed:"
+    prompt = f"Fix this malformed LaTeX expression so it's valid:\n{data.latex}"
     try:
-        result = pipe(prompt, do_sample=False)[0]["generated_text"]
-        fixed = result.split("Fixed:")[-1].strip("` \n")
-        return JSONResponse(content={"fixed_latex": fixed})
+        result = pipe(prompt)[0]["generated_text"]
+        return JSONResponse(content={"fixed_latex": result.strip()})
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)