Update app.py

app.py

@@ -48,7 +48,12 @@ async def extract_text(file: UploadFile = File(...)):
         contents = await file.read()
         image = Image.open(io.BytesIO(contents)).convert("RGB")
 
-        inputs = processor(images=image, return_tensors="pt")
+        # inputs = processor(images=image, return_tensors="pt")
+        inputs = processor(
+        text="Extract all text from the document",
+        images=image,
+        return_tensors="pt"
+)
 
         with torch.no_grad():
             outputs = model.generate(**inputs, max_new_tokens=1024)