Update app.py

app.py

@@ -8,9 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_cod
 model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', 
                                   trust_remote_code=True, 
                                   low_cpu_mem_usage=True, 
-                                  device_map='cuda', 
-                                  use_safetensors=True, 
-                                  pad_token_id=tokenizer.eos_token_id).eval().cuda()
+                                  pad_token_id=tokenizer.eos_token_id).eval()  # Removed device_map='cuda' and .cuda()
 
 # Define the function to process images and extract text
 def extract_text_from_image(image):