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()