File size: 565 Bytes
31a04a4 39e5b56 041a59d 31a04a4 041a59d 31a04a4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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() |