AWS-ANSWER-BOT commited on
Commit
37412c9
·
verified ·
1 Parent(s): 301c7b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Load a conversational model from Hugging Face Hub
5
- chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
6
 
7
  def respond(message, chat_history):
8
  chat_history = chat_history or []
 
2
  from transformers import pipeline
3
 
4
  # Load a conversational model from Hugging Face Hub
5
+ chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
6
 
7
  def respond(message, chat_history):
8
  chat_history = chat_history or []