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

def predict(prompt):
    completion = model(prompt)[0]["generated_text"]
    return completion

demo = gr.Interface(fn=predict, inputs="text", outputs="text")
demo.launch()