Update app.py

app.py

@@ -13,10 +13,10 @@ from datasets import load_dataset, Image
 from PIL import Image
 from paddleocr import PaddleOCR
 from doctr.io import DocumentFile
-from doctr.models import ocr_predictor
-
 
 
+os.environ["USE_TORCH"] = "1"
+from doctr.models import ocr_predictor
 ocr_model = ocr_predictor(pretrained=True)