Update app.py

app.py

@@ -12,16 +12,15 @@ import gradio as gr
 from fastapi.responses import RedirectResponse
 import numpy as np
 import easyocr
-import os
+
 # Initialize FastAPI
 app = FastAPI()
 
-print(f"🔄 Loading models")
-model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
-hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")  # Fetch token from environment variable
-
-tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
-model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token)
+# Load AI Model for Question Answering on Documents (Mistral-7B)
+model_name = "TinyLlama/TinyLlama-1.1B"
+print(f"🔄 Loading model: {model_name}...")
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 doc_qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=-1)