Update app.py

app.py

@@ -1,8 +1,9 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
- # Load pretrained model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("flan-t5-large")
-model = AutoModelForCausalLM.from_pretrained("flan-t5-large")
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+
+
+model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
+tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
 
 def chatbot_model(prompt):