m

Dockerfile

@@ -27,7 +27,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY --chown=user . .
 
 # Expose the port FastAPI will run on
-EXPOSE 7860
+EXPOSE 7860 8000
 
 # Command to run the FastAPI application
 CMD ["python", "run.py"]

main.py

@@ -31,6 +31,11 @@ class QueryResponse(BaseModel):
     similar_prompts: List[SimilarPrompt]
 
 
+@app.get("/")
+def read_root():
+    return {"message": "Hello, FastAPI!"}
+
+
 # API endpoint
 @app.post("/most_similar", response_model=QueryResponse)
 async def get_most_similar(query_request: QueryRequest):

run.py

@@ -3,5 +3,5 @@ import uvicorn
 # import subprocess
 
 if __name__ == "__main__":
-    uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=True)
+    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)
     # subprocess.run(["streamlit", "run", "app.py", "--server.port", "7860"])