Spaces:
Sleeping
Sleeping
update
Browse files- Dockerfile +1 -1
- 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 |
+
|