import gradio as gr # Load the model model = gr.load("models/dataautogpt3/ProteusV0.2") # Customize the interface iface = gr.Interface( fn=model, title="", description="", inputs="text", outputs="image", # Set outputs to "image" for image output ) # Launch the customized interface iface.launch()