Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Create a class for the session state
 class SessionState:
@@ -18,7 +18,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 = "llmware/bling-red-pajamas-3b-0.1"
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Create a chat input for the user