Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
@@ -18,7 +17,7 @@ temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.7)
 
 # Load the model and tokenizer with a loading message
 with st.spinner('Wait for model to respond..'):
-    model_name = "facebook/blenderbot-3B"
+    model_name = "llmware/bling-red-pajamas-3b-0.1"
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)