Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
-from transformers import InferenceClient
+from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+from PIL import Image
 
 client = InferenceClient("models/microsoft/trocr-base-handwritten")
 
@@ -20,11 +21,13 @@ def sepia(input_img):
 ## required positional arguments: 'inputs' and 'outputs'
 def process_image(image):
     try:
-        # Run the image through the model
-        result = client(inputs=image)
-        # Extract the text from the result
-        text = result['text']
-        return text
+        processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
+        model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
+        pixel_values = processor(images=image, return_tensors="pt").pixel_values
+        
+        generated_ids = model.generate(pixel_values)
+        generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+        return generated_text
     except Exception as e:
         return f"Error: {str(e)}"