new0KWR commited on
Commit
c24b784
·
verified ·
1 Parent(s): 6b9ca5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
5
 
6
  def chat_with_bot(message):
7
  return chatbot(message)[0]['generated_text']
8
 
9
  gr.Interface(fn=chat_with_bot, inputs="text", outputs="text").launch()
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ chatbot = pipeline("conversational", model="dkleczek/bert-base-polish-cased-v1")
5
 
6
  def chat_with_bot(message):
7
  return chatbot(message)[0]['generated_text']
8
 
9
  gr.Interface(fn=chat_with_bot, inputs="text", outputs="text").launch()
10
+