Spaces:
Sleeping
Sleeping
Nikhil0987
commited on
Commit
•
907ef02
1
Parent(s):
4d2da6c
Update convo.py
Browse files
convo.py
CHANGED
@@ -2,10 +2,11 @@ from transformers import pipeline, Conversation
|
|
2 |
import streamlit as st
|
3 |
from streamlit_option_menu import option_menu
|
4 |
|
|
|
5 |
|
6 |
def Convo():
|
7 |
-
convo
|
8 |
-
|
9 |
conversation = Conversation(convo)
|
10 |
|
11 |
|
|
|
2 |
import streamlit as st
|
3 |
from streamlit_option_menu import option_menu
|
4 |
|
5 |
+
chatbot = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
|
6 |
|
7 |
def Convo():
|
8 |
+
if convo := st.chat_input("Enter your message")
|
9 |
+
|
10 |
conversation = Conversation(convo)
|
11 |
|
12 |
|