import gradio as gr
from transformers import pipeline
export TF_ENABLE_ONEDNN_OPTS=0

pipeline = pipeline(task="text-generation", model="Preetham04/text-generation")

def predict(input_img):
    predictions = pipeline(input_img)
    return  {p["labels"] for p in predictions}

gradio_app = gr.Interface(
    predict,
    inputs="textbox", 
    outputs="text",
    title="Text-generation",
)

if __name__ == "__main__":
    gradio_app.launch(share=True)