import gradio as gr
from transformers import pipeline
pipe = pipeline("text-generation", model="succinctly/text2image-prompt-generator")
def engineer(input):
    return pipe(input)[0]["generated_text"]

iface = gr.Interface(title="Chameleon Prompt Engineering AI 🦎", fn=engineer, inputs="text", outputs="text")
iface.launch()