Update app.py

app.py

@@ -16,15 +16,11 @@ import easyocr
 # Initialize FastAPI
 app = FastAPI()
 
-# Load AI Model for Question Answering on Documents (Mistral-7B)
-model_name = "TinyLlama/TinyLlama-1.1B-chat"
 print(f"🔄 Loading models")
 
-doc_qa_pipeline = pipeline("text-generation", model=model)
-
-# Load Image Captioning Model (nlpconnect/vit-gpt2-image-captioning)
+doc_qa_pipeline = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-chat")
 image_captioning_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
-
+print("models loaded")
 # Initialize OCR Model (Lazy Load)
 reader = easyocr.Reader(["en"], gpu=True)