Update app.py

app.py

@@ -1,22 +1,19 @@
 from fastapi import FastAPI, UploadFile, File
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from PIL import Image
-import torch
 
 app = FastAPI()
 
-# Load TrOCR model + processor
-processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-stage1")
-model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-stage1")
+try:
+    processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-stage1")
+    model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-stage1")
+except Exception as e:
+    print("Model load failed:", e)
 
 @app.post("/predict")
 async def predict(file: UploadFile = File(...)):
-    # open uploaded image
     image = Image.open(file.file).convert("RGB")
-
-    # run OCR
     pixel_values = processor(image, return_tensors="pt").pixel_values
     generated_ids = model.generate(pixel_values)
     text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-
     return {"text": text}