Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
 @st.cache_data
 def load_model(model_name):
-    model = AutoModelForSeq2SeqLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
+    model = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
     return model
 
 model = load_model("mistralai/Mixtral-8x7B-Instruct-v0.1")