Update app.py

app.py

@@ -2,12 +2,10 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
-import torch
 
 # Initialize the model and processor
 processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
 model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwritten")
-model.eval()
 
 def display_sketch(sketch):
     if isinstance(sketch, dict) and 'composite' in sketch:
@@ -38,8 +36,14 @@ with gr.Blocks() as demo:
     output_image = gr.Image(label="Your Sketch")
     recognized_text = gr.Textbox(label="Recognized Text")
     
-    sketchpad_output = sketchpad.change(display_sketch, inputs=sketchpad, outputs=output_image)
-    sketchpad_output.change(recognize_text, inputs=output_image, outputs=recognized_text)
+    def process_and_recognize(sketch):
+        image_path = display_sketch(sketch)
+        if image_path:
+            text = recognize_text(image_path)
+            return image_path, text
+        return None, ""
+
+    sketchpad.change(process_and_recognize, inputs=sketchpad, outputs=[output_image, recognized_text])
 
 # Launch the demo
 demo.launch()