Update app.py

app.py

@@ -1,12 +1,12 @@
 # URL: https://huggingface.co/spaces/gradio/text_generation
 # imports
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import GPT2Tokenizer, TFGPT2Model
 import torch
 
 # loading the model
-tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
-model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-j-6B")
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+model = TFGPT2Model.from_pretrained('gpt2')
 
 # defining the core function
 def generate(text):
@@ -15,9 +15,8 @@ def generate(text):
  return result[0]["generated_text"]
 
 
-# defining title, description and examples
-title = "Text Generation with GPT-J-6B"
-description = "This demo generates text using GPT-J 6B: a transformer model trained using Ben Wang's Mesh Transformer JAX."
+# defining title and examples
+title = "Text Generation with GPT-2"
 examples = [
  ["The Moon's orbit around Earth has"],
  ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],