File size: 675 Bytes
e36951d 673420a e36951d 673420a e36951d 673420a 51b28a6 673420a e36951d 673420a e36951d 673420a e36951d 673420a 08185a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import streamlit as st
from transformers import pipeline
# Load Hugging Face chatbot model
chatbot_model = pipeline("conversational", model="alpindale/goliath-120b")
# Main Page with Chatbot
def main():
st.title("Virtual Therapist Chatbot")
st.write("Feel free to chat with our virtual therapist!")
# User input for the chatbot
user_input = st.text_input("You: ")
# Generate response when user enters input
if user_input:
response = chatbot_model(user_input, max_length=50, num_return_sequences=1)[0]['generated_text']
st.text_area("Therapist:", response, height=100)
# Run the main function
if __name__ == "__main__":
main()
|