Update app.py

app.py

@@ -87,7 +87,7 @@ templates = Jinja2Templates(directory="templates")
 import os
 huggingface_token = os.environ.get("HUGGINGFACE_TOKEN")
 # Download and load the model
-model2 = VisionEncoderDecoderModel.from_pretrained("musadac/vilanocr-single-urdu",use_auth_token=huggingface_token).to(device)
+model2 = VisionEncoderDecoderModel.from_pretrained("musadac/vilanocr-single-urdu",use_auth_token=huggingface_token)
 
 
 @app.get("/", response_class=HTMLResponse)
@@ -98,13 +98,13 @@ async def root():
 async def upload_image(image: UploadFile = File(...)):
     # Preprocess image
     img = Image.open(image.file).convert("RGB")
-    pixel_values = processortext(img.convert("RGB"), return_tensors="pt").pixel_values
+    pixel_values = processortext2(img.convert("RGB"), return_tensors="pt").pixel_values
     
     # Run the model
     with torch.no_grad():
         generated_ids = model2.generate(img_tensor)
     
     # Extract OCR result
-    result = processortext.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    result = processortext2.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
     return {"result": result}