Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 import json
+from transformers import AutoTokenizer, AutoModelWithLMHead
+  
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
 
-def green(t):
-    pass
+model = AutoModelWithLMHead.from_pretrained("gpt2")
+def green(prompt):
+    print(tokenizer("help"))
 iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs="text")
 iface.launch()