| import gradio as gr | |
| from transformers import pipeline | |
| from diffusers import DiffusionPipeline | |
| pipeline = DiffusionPipeline.from_pretrained("cloudqi/cqi_text_to_image_pt_v0") | |
| inputs = gr.inputs.Textbox(prompt="Enter Text Description") | |
| outputs = gr.outputs.Image(label="Generated Image") | |
| iface = gr.Interface(fn=generate_images, inputs=inputs, outputs=outputs, title="Description to Image") | |
| iface.disable_caching=True # Disable caching to ensure the model is reloaded each time the app is opened | |
| if __name__ == "__main__": | |
| iface.launch() |