Update app.py

app.py

@@ -1,10 +1,16 @@
 import gradio as gr
 from datasets import load_dataset
+from transformers import pipeline
 
 dataset = load_dataset("rebase/hello")
+model = pipeline("text-generation")
 
 def image_classifier(inp):
     return dataset
 
-demo = gr.Interface(fn=image_classifier, inputs="text", outputs="text")
+def predict(prompt):
+    completion = model(prompt)[0]["generated_text"]
+    return completion
+
+demo = gr.Interface(fn=predict, inputs="text", outputs="text")
 demo.launch()