Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 import torch
-import spaces
 import subprocess
 import json
 from PIL import Image, ImageDraw
@@ -38,11 +37,11 @@ def load_model(model_name):
         current_model_name = model_name
         
         # Move model to GPU
-        current_model = current_model.to('cuda')
+        # current_model = current_model.to('cuda')
     
     return current_processor, current_model
 
-@spaces.GPU
+
 def process_image(image, model_name):
     # Save the uploaded image to a temporary file
     with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as temp_img:
@@ -72,7 +71,7 @@ def process_image(image, model_name):
 
         # Prepare image for TrOCR
         pixel_values = processor(line_image, return_tensors="pt").pixel_values
-        pixel_values = pixel_values.to('cuda')
+        # pixel_values = pixel_values.to('cuda')
 
         # Generate (no beam search)
         with torch.no_grad():