Upload scripts/api.py with huggingface_hub

scripts/api.py

@@ -30,6 +30,10 @@ class QueryResponse(BaseModel):
 def root():
     return {"message": "Welcome to the Prompt Search API. Use '/search' endpoint to find similar prompts."}
 
+def start_api_server():
+    import uvicorn
+    uvicorn.run(app, host = config["server"]["host"], port = int(os.getenv("PORT", config["server"]["port"])))
+
 @app.post("/search", response_model=QueryResponse)
 async def search_prompts(query_request: QueryRequest):
     """
@@ -60,8 +64,3 @@ async def search_prompts(query_request: QueryRequest):
     except Exception as e:
         print(e)
         raise HTTPException(status_code=500, detail=str(e))
-
-# Entry point for configuring parameters and running the app
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host = config["server"]["host"], port = int(os.getenv("PORT", config["server"]["port"])))