Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from fastapi import FastAPI, UploadFile, File, HTTPException
|
| 2 |
from fastapi.responses import HTMLResponse
|
| 3 |
from huggingface_hub import InferenceClient
|
| 4 |
from transformers import pipeline
|
|
@@ -44,7 +44,7 @@ HTML_CONTENT = """
|
|
| 44 |
body: JSON.stringify({ "prompt": prompt })
|
| 45 |
});
|
| 46 |
let data = await response.json();
|
| 47 |
-
responseBox.innerText = data.text;
|
| 48 |
}
|
| 49 |
</script>
|
| 50 |
</head>
|
|
@@ -95,7 +95,7 @@ async def generate_text(prompt: str) -> str:
|
|
| 95 |
return "Xin lỗi, tôi gặp lỗi khi xử lý câu hỏi của bạn."
|
| 96 |
|
| 97 |
@app.post("/chatbot")
|
| 98 |
-
async def chatbot(prompt: str = None, file: UploadFile = None):
|
| 99 |
try:
|
| 100 |
if file:
|
| 101 |
stt_output = await speech_to_text(file)
|
|
@@ -114,4 +114,4 @@ async def chatbot(prompt: str = None, file: UploadFile = None):
|
|
| 114 |
|
| 115 |
if __name__ == "__main__":
|
| 116 |
print("🚀 Khởi động FastAPI Server...")
|
| 117 |
-
uvicorn.run(app, host="0.0.0.0", port=7860)
|
|
|
|
| 1 |
+
from fastapi import FastAPI, UploadFile, File, HTTPException, Form
|
| 2 |
from fastapi.responses import HTMLResponse
|
| 3 |
from huggingface_hub import InferenceClient
|
| 4 |
from transformers import pipeline
|
|
|
|
| 44 |
body: JSON.stringify({ "prompt": prompt })
|
| 45 |
});
|
| 46 |
let data = await response.json();
|
| 47 |
+
responseBox.innerText = data.text || "Lỗi xảy ra";
|
| 48 |
}
|
| 49 |
</script>
|
| 50 |
</head>
|
|
|
|
| 95 |
return "Xin lỗi, tôi gặp lỗi khi xử lý câu hỏi của bạn."
|
| 96 |
|
| 97 |
@app.post("/chatbot")
|
| 98 |
+
async def chatbot(prompt: str = Form(None), file: UploadFile = File(None)):
|
| 99 |
try:
|
| 100 |
if file:
|
| 101 |
stt_output = await speech_to_text(file)
|
|
|
|
| 114 |
|
| 115 |
if __name__ == "__main__":
|
| 116 |
print("🚀 Khởi động FastAPI Server...")
|
| 117 |
+
uvicorn.run(app, host="0.0.0.0", port=7860)
|