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() |