changes in api

api.py

@@ -19,7 +19,6 @@ class QueryResponse(BaseModel):
 # Initialize FastAPI app
 app = FastAPI()
 
-# Global variable to store the search engine
 search_engine = None
 
 # Load prompts and initialize the search engine when the app starts
@@ -29,8 +28,6 @@ def startup_event():
     # Load the prompts
     dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
     prompts = dataset["train"]["Prompt"]
-    # For testing, limit the number of prompts
-    prompts = prompts[:1000]  # Adjust the number as needed
     # Initialize vectorizer with the default model
     vectorizer = Vectorizer(model="all-MiniLM-L6-v2")
     # Initialize the search engine