import torch
import gradio as gr


def doo(text_prompt):
    return 

iface = gr.Interface(
    fn=doo,
    inputs=gr.Textbox(label="Prompt"),
    outputs=[
        gr.Image(label="Generated Image"),
    ] 
)


if __name__ == "__main__":
    iface.launch()