Update app.py

app.py

@@ -119,13 +119,13 @@
 
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, GenerationConfig
+from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 import re
 import time
 
 
 tokenizer = AutoTokenizer.from_pretrained("universitytehran/PersianMind-v1.0")
-model = AutoModelForSeq2SeqLM.from_pretrained("universitytehran/PersianMind-v1.0")
+model = AutoModelForCausalLM.from_pretrained("universitytehran/PersianMind-v1.0")
 
 
 def generate_response(message, chat_history):