Update app.py

app.py

@@ -17,8 +17,21 @@ import easyocr
 app = FastAPI()
 
 # Load AI Model for Question Answering (DeepSeek-V2-Chat)
-qa_pipeline = pipeline("text-generation", model="microsoft/phi-2", device=0 if torch.cuda.is_available() else -1)
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
+# Preload Hugging Face model
+model_name = "microsoft/phi-2"
+print(f"🔄 Loading model: {model_name}...")
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+qa_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=device)
+
+# Load Pretrained Object Detection Model (Torchvision)
+from torchvision.models.detection import FasterRCNN_ResNet50_FPN_Weights
+weights = FasterRCNN_ResNet50_FPN_Weights.DEFAULT
+model = fasterrcnn_resnet50_fpn(weights=weights)
+model.eval()
 # Load Pretrained Object Detection Model (if needed)
 model = fasterrcnn_resnet50_fpn(pretrained=True)
 model.eval()