Update app.py

app.py

@@ -4,9 +4,9 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from PIL import Image
 from sympy import sympify, solve, Eq, symbols
 
-# Load the math OCR model and processor
-processor = TrOCRProcessor.from_pretrained("nlpai-lab/mathocr-htr-base")
-model = VisionEncoderDecoderModel.from_pretrained("nlpai-lab/mathocr-htr-base")
+# Load the math OCR model and processor (publicly available)
+processor = TrOCRProcessor.from_pretrained("lambdalabs/smicr_ocr_exp1")
+model = VisionEncoderDecoderModel.from_pretrained("lambdalabs/smicr_ocr_exp1")
 
 def predict_math_problem(image):
     try:
@@ -16,7 +16,7 @@ def predict_math_problem(image):
         generated_ids = model.generate(pixel_values)
         transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         
-        # Standardize mathematical symbols in the transcription
+        # Standardize mathematical symbols
         transcription = (transcription
             .replace("×", "*")
             .replace("÷", "/")
@@ -26,13 +26,12 @@ def predict_math_problem(image):
             .replace("³", "**3")
             .replace("½", "1/2")
             .replace("¼", "1/4")
-            .replace("…", "...")  # Ellipsis
+            .replace("…", "...")
         )
         
-        # Attempt to solve the mathematical problem
+        # Solve the mathematical expression
         solution = None
         try:
-            # Check if the transcription is an equation (contains '=')
             if '=' in transcription:
                 lhs, rhs = transcription.split('=', 1)
                 equation = Eq(sympify(lhs.strip()), sympify(rhs.strip()))
@@ -42,29 +41,27 @@ def predict_math_problem(image):
                     solution = solve(equation, variable)
                     solution = f"{variable} = {solution}"
                 else:
-                    solution = "No variables found in equation"
+                    solution = "Solution: No Variables Found"
             else:
-                # Treat as an arithmetic expression
-                solution = sympify(transcription)
-                solution = f"Result: {solution}"
+                solution = f"Result: {sympify(transcription)}"
         except:
-            solution = "Invalid or unsolvable expression"
+            solution = "Error: Unable to Solve Expression"
         
         return transcription, solution
     
     except Exception as e:
-        return f"Error: {str(e)}", "Failed to process"
+        return f"Error: {str(e)}", "Processing Failed"
 
 # Create Gradio interface
 demo = gr.Interface(
     fn=predict_math_problem,
-    inputs=gr.Image(type="pil", label="Upload Handwritten Math Problem"),
+    inputs=gr.Image(type="pil", label="Handwritten Math Problem"),
     outputs=[
-        gr.Textbox(label="Transcribed Text"),
-        gr.Textbox(label="Solution")
+        gr.Textbox(label="Transcribed Math Text"),
+        gr.Textbox(label="Math Solution")
     ],
     title="Handwritten Math Solver",
-    description="Upload a handwritten math problem to get its transcription and solution."
+    description="Upload a handwritten math image to get transcription and solution."
 )
 
 if __name__ == "__main__":