Update app.py

app.py

@@ -9,11 +9,15 @@ model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwri
 
 @spaces.GPU
 def recognize_text(image):
-    image = image.convert("RGB")
-    pixel_values = processor(images=image, return_tensors="pt").pixel_values
-    generated_ids = model.generate(pixel_values)
-    text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    return text
+    try:
+        image = image.convert("RGB")
+        pixel_values = processor(images=image, return_tensors="pt").pixel_values
+        generated_ids = model.generate(pixel_values)
+        text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+        
+        return text
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 # Gradio UI
 note = gr.Interface(
@@ -24,4 +28,4 @@ note = gr.Interface(
     description="Upload an image of handwritten text, and the AI will convert it to digital text."
 )
 
-note.launch()
+note.launch()