Update app.py

app.py

@@ -1,4 +1,12 @@
+from transformers import pipeline
 import gradio as gr
-name_list = ['models/gpt2']
-interfaces = [gr.Interface.load(name) for name in name_list]
-gr.mix.Parallel(*interfaces, title="GPT2 tester", description="GPT2 tester GPT2 tester").launch()
+
+# First define a prediction function that takes in a text prompt and returns the text completion
+model = pipeline('text-generation')
+
+def predict(prompt):
+    completion = model(prompt)[0]['generated_text']
+    return completion
+
+# Now create the interface
+gr.Interface(fn=predict, inputs="text", outputs="text").launch()