Update app.py

app.py

@@ -17,12 +17,10 @@ import easyocr
 app = FastAPI()
 
 # Load AI Model for Question Answering on Documents (Mistral-7B)
-model_name = "meta-llama/Llama-2-7b-chat-hf"
-print(f"🔄 Loading model: {model_name}...")
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model_name = "TinyLlama/TinyLlama-1.1B-chat"
+print(f"🔄 Loading models")
 
-doc_qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=-1)
+doc_qa_pipeline = pipeline("text-generation", model=model)
 
 # Load Image Captioning Model (nlpconnect/vit-gpt2-image-captioning)
 image_captioning_pipeline = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")