Update app.py

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 import numpy as np
 from PIL import Image
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
-import matplotlib.pyplot as plt
-import tempfile
+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,9 +38,8 @@ with gr.Blocks() as demo:
     output_image = gr.Image(label="Your Sketch")
     recognized_text = gr.Textbox(label="Recognized Text")
     
-    sketchpad.submit(display_sketch, inputs=sketchpad, outputs=output_image).then(
-        recognize_text, inputs=output_image, outputs=recognized_text
-    )
+    sketchpad_output = sketchpad.change(display_sketch, inputs=sketchpad, outputs=output_image)
+    sketchpad_output.change(recognize_text, inputs=output_image, outputs=recognized_text)
 
 # Launch the demo
 demo.launch()