Update app.py

app.py

@@ -2,18 +2,26 @@ from fastapi import FastAPI, UploadFile, File
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from PIL import Image
 
+print("🚀 App is starting...")
+
 app = FastAPI()
 
 try:
+    print("⏳ Loading TrOCR model...")
     processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-stage1")
     model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-stage1")
+    print("✅ Model loaded successfully")
 except Exception as e:
-    print("Model load failed:", e)
+    print("❌ Model load failed:", e)
 
 @app.post("/predict")
 async def predict(file: UploadFile = File(...)):
+    print("📸 Received image, running OCR...")
     image = Image.open(file.file).convert("RGB")
+
     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]
+
+    print("🔤 OCR Result:", text)
     return {"text": text}