Update main.py

main.py

@@ -6,7 +6,7 @@ from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from transformers import pipeline
 from PIL import Image
-from vector import query_vector
+from smebuilder_vector import query_vector
 
 # ==============================
 # Logging Setup