rebase/hello_python

1import gradio as gr2#from datasets import load_dataset3from transformers import pipeline4 5#dataset = load_dataset("rebase/hello")6model = pipeline("text-generation")7 8def image_classifier(inp):9    return "dataset"10 11def predict(prompt):12    completion = model(prompt)[0]["generated_text"]13    return completion14 15demo = gr.Interface(fn=predict, inputs="text", outputs="text")16demo.launch()17