Update app.py

app.py

@@ -50,6 +50,7 @@ def run_ocr(image: Image.Image) -> str:
     
     # Generate text from the image
     # Note: max_new_tokens can be adjusted based on expected text length
+    os.environ["DISABLE_FLASH_ATTN"] = "1"
     generated_ids = model.generate(
         input_ids=inputs["input_ids"],
         pixel_values=inputs["pixel_values"],