Update main.py

main.py

@@ -6,7 +6,7 @@ from vllm import LLM
 import numpy as np
 
 # Initialize the model
-llm = LLM(model='BAAI/bge-base-en-v1.5', task="embed")
+llm = LLM(model='BAAI/bge-base-en-v1.5', task="embed", device='cpu')
 
 # Initialize FastAPI app
 app = FastAPI()