Update app.py

app.py

@@ -17,7 +17,7 @@ import easyocr
 app = FastAPI()
 
 # Load AI Model for Question Answering (DeepSeek-V2-Chat)
-qa_pipeline = pipeline("text-generation", model="microsoft/phi-2")
+qa_pipeline = pipeline("text-generation", model="microsoft/phi-2", device=0 if torch.cuda.is_available() else -1)
 
 # Load Pretrained Object Detection Model (if needed)
 model = fasterrcnn_resnet50_fpn(pretrained=True)