kmondal91 commited on
Commit
1d61154
·
verified ·
1 Parent(s): d1e5142

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from transformers import pipeline
3
 
4
  # Load the Hugging Face pipeline
5
- chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
6
 
7
  st.set_page_config(page_title="Chatbot", layout="centered")
8
 
 
2
  from transformers import pipeline
3
 
4
  # Load the Hugging Face pipeline
5
+ chatbot = pipeline("text-generation", model="microsoft/Phi-4-mini-reasoning")
6
 
7
  st.set_page_config(page_title="Chatbot", layout="centered")
8