Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import GPTJForCausalLM, GPT2Tokenizer
 
 # Load model and tokenizer
-model_name = "gpt2"
-model = GPT2LMHeadModel.from_pretrained(model_name)
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+model_name = "EleutherAI/gpt-j-6B"
+model = GPTJForCausalLM.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)  # GPT-J uses the same tokenizer as GPT-2
 
 # Function to filter explicit content
 def filter_explicit(content, filter_on):