fixed cuda bug for pixel values

app.py

@@ -95,15 +95,19 @@ def visualize_lines(image, lines):
 @spaces.GPU
 def transcribe_lines(line_images, model_name):
     processor, model = load_model(model_name)
-    
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     transcriptions = []
     for line_image in line_images:
         # Process the line image
         pixel_values = processor(images=line_image, return_tensors="pt").pixel_values
 
+        pixel_values = pixel_values.to(device)
+
         # Generate (no beam search)
         generated_ids = model.generate(pixel_values)
 
+        
+
         # Decode
         generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         transcriptions.append(generated_text)