freud-sensei commited on
Commit
7c980a6
1 Parent(s): 855cbd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- def counsel_chat(message):
5
  pipeline_model = pipeline("text-generation", model="freud-sensei/counsel_chatbot")
6
  sentence = f"질문: {message} 답변:"
7
  answer = pipeline_model(sentence)
8
  return answer[0]["generated_text"][len(sentence):]
9
 
10
- chatbot2 = gr.Interface(counsel_chat, "text", "text")
11
  chatbot2.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ def counsel_chat(message, history):
5
  pipeline_model = pipeline("text-generation", model="freud-sensei/counsel_chatbot")
6
  sentence = f"질문: {message} 답변:"
7
  answer = pipeline_model(sentence)
8
  return answer[0]["generated_text"][len(sentence):]
9
 
10
+ chatbot2 = gr.ChatInterface(counsel_chat)
11
  chatbot2.launch()