Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 import torch
 from torchvision import transforms
 import matplotlib.pyplot as plt
+import spaces
 
 # Load TrOCR model
 processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
@@ -28,6 +29,7 @@ def visualize_image(pixel_values):
     plt.title("Preprocessed Image")
     plt.show()
 
+@spaces.GPU
 def recognize_text(image):
     try:
         # Preprocess the image