Update app.py

app.py

@@ -1,11 +1,18 @@
 import gradio as gr
 import json
 from transformers import AutoTokenizer, AutoModelWithLMHead
-  
+from transformers import pipeline
 tokenizer = AutoTokenizer.from_pretrained("gpt2")
-
+model = AutoModelWithLMHead.from_pretrained("gpt2-large")
+generator = pipeline(
+  task="text-generation",
+  model=model,
+  tokenizer=tokenizer,
+  framework="tf"
+)
 model = AutoModelWithLMHead.from_pretrained("gpt2")
 def green(prompt):
-    print(tokenizer("help"))
+    text_generator(prompt, max_length=500)
+    
 iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs="text")
 iface.launch()