deadshot2003 commited on
Commit
0564d5c
1 Parent(s): 8ba39ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,13 +8,13 @@ import uvicorn
8
 
9
  app = FastAPI()
10
 
11
- chatbot = pipeline(model="bluenguyen/movie_chatbot_large_v1")
12
 
13
  def DS_chatbot(message,history):
14
  conversation = chatbot(message)
15
  return conversation[0]['generated_text']
16
 
17
- io = gr.ChatInterface(DS_chatbot, title=" DS Chatbot", description="Enter text to start chatting.")
18
  @app.get("/")
19
  async def read_main():
20
  return {"message": "Append /gradio to the url to see gradio the interface"
 
8
 
9
  app = FastAPI()
10
 
11
+ chatbot = pipeline(model="Kaludi/Customer-Support-Assistant-V2")
12
 
13
  def DS_chatbot(message,history):
14
  conversation = chatbot(message)
15
  return conversation[0]['generated_text']
16
 
17
+ io = gr.ChatInterface(DS_chatbot, title=" Customer Service Bot", description="Enter your query.")
18
  @app.get("/")
19
  async def read_main():
20
  return {"message": "Append /gradio to the url to see gradio the interface"