Update app.py

app.py

@@ -2,9 +2,7 @@ from fastapi import FastAPI
 from fastapi.responses import RedirectResponse
 import gradio as gr
 from PIL import Image
-import io
 import numpy as np
-import easyocr
 from transformers import pipeline
 from gtts import gTTS
 import tempfile
@@ -12,8 +10,8 @@ import os
 
 app = FastAPI()
 
-# OCR Reader
-ocr_reader = easyocr.Reader(['en'], gpu=False)
+# OCR Reader (lazy import inside function to avoid ImportError on Spaces)
+ocr_reader = None
 
 # Captioning and VQA Pipelines
 caption_model = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
@@ -24,6 +22,12 @@ def process_image_question(image: Image.Image, question: str):
         return "No image uploaded.", None
 
     try:
+        # Import EasyOCR only when needed
+        global ocr_reader
+        if ocr_reader is None:
+            import easyocr
+            ocr_reader = easyocr.Reader(['en'], gpu=False)
+
         # Convert PIL image to numpy array
         np_image = np.array(image)