tudeplom commited on
Commit
043d9c7
·
verified ·
1 Parent(s): d83895a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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)