Update app.py

app.py

@@ -1,28 +1,26 @@
 import gradio as gr
-from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+import torch
+from transformers import VisionEncoderDecoderModel, TrOCRProcessor
 
 # Load the model and processor
 processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
 model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
 
-def detect_objects(image):
-    # Preprocess the image
+# Define a function to perform OCR
+def perform_ocr(image):
     pixel_values = processor(images=image, return_tensors="pt").pixel_values
-    
-    # Generate the text
     generated_ids = model.generate(pixel_values)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    
     return generated_text
 
-# Create the Gradio interface
-gr_interface = gr.Interface(
-    fn=detect_objects,
+# Create a Gradio interface
+iface = gr.Interface(
+    fn=perform_ocr,
     inputs="image",
     outputs="text",
     title="Image OCR",
-    description="Scan images into text format"
+    description="Upload an image to extract text"
 )
 
 # Launch the interface
-gr_interface.launch()
+iface.launch()