Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import time
 
@@ -68,7 +68,7 @@ def load_model():
     # Using VietAI's Vietnamese GPT model
     model_name = "google-t5/t5-base"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     return model, tokenizer
 
 def generate_response(prompt, model, tokenizer, max_length=4169):