Update app.py

app.py

@@ -119,7 +119,7 @@ MODEL_ID_3B = "nanonets/Nanonets-OCR2-3B"
 processor_3b = AutoProcessor.from_pretrained(MODEL_ID_3B, trust_remote_code=True)
 model_3b = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_3B,
-    dtype=torch.float16,
+    torch_dtype=torch.float16,
     trust_remote_code=True,
 ).to(device).eval()
 
@@ -128,7 +128,7 @@ MODEL_ID_1_5B = "nanonets/Nanonets-OCR2-1.5B-exp"
 processor_1_5b = AutoProcessor.from_pretrained(MODEL_ID_1_5B, trust_remote_code=True)
 model_1_5b = AutoModelForImageTextToText.from_pretrained(
     MODEL_ID_1_5B,
-    dtype=torch.float16,
+    torch_dtype=torch.float16,
     trust_remote_code=True,
     attn_implementation="flash_attention_2"
 ).to(device).eval()