arcsu1 commited on
Commit
031cff7
·
1 Parent(s): cfe1b6a
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. main.py +4 -2
Dockerfile CHANGED
@@ -16,4 +16,4 @@ RUN pip install --no-cache-dir -r requirements.txt
16
  RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
17
 
18
  # Start the FastAPI app on port 7860, the default port expected by Spaces
19
- CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
16
  RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
17
 
18
  # Start the FastAPI app on port 7860, the default port expected by Spaces
19
+ CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "4"]
main.py CHANGED
@@ -27,7 +27,7 @@ class HistoryRequest(BaseModel):
27
  ai: list[str]
28
 
29
  @app.post("/generate")
30
- def generate_response(history: HistoryRequest):
31
  try:
32
  model
33
  except:
@@ -37,4 +37,6 @@ def generate_response(history: HistoryRequest):
37
  model = chatbot.ChatBot()
38
 
39
  response = model.generate_response(history)
40
- return response
 
 
 
27
  ai: list[str]
28
 
29
  @app.post("/generate")
30
+ async def generate_response(history: HistoryRequest):
31
  try:
32
  model
33
  except:
 
37
  model = chatbot.ChatBot()
38
 
39
  response = model.generate_response(history)
40
+ return response
41
+
42
+