use GPT2 instead of T5

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import numpy as np
 
 
-MODEL_NAME = "google/flan-t5-base"
+MODEL_NAME = "gpt2"
 
 
 if __name__ == "__main__":
     # Define your model and your tokenizer
     tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)  # or AutoModelForCausalLM
+    model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)  # or AutoModelForCausalLM
     if tokenizer.pad_token_id is None:
         tokenizer.pad_token_id = tokenizer.eos_token_id
         model.config.pad_token_id = model.config.eos_token_id