| import gradio as gr |
| from huggingface_hub import hf_hub_download |
| from safetensors.torch import load_file |
| from PIL import Image |
|
|
| from model import * |
|
|
| def generate_image(prompt): |
| |
| return prompt_to_img(prompt)[0] |
|
|
|
|
|
|
| |
| description = """ |
| This demo utilizes a specialized variant of the Stable Diffusion model designed for multilingual text-to-image synthesis. In response to the observed underperformance of existing models on languages beyond English, this project introduces the Multilingual Stable Diffusion, providing a more inclusive solution for diverse linguistic contexts. |
| Link to Github repo: https://github.com/NajlaaNawaii/Multilingual-Stable-Diffusion-Towards-more-Inclusive-Text-To-Image-Synthesis |
| """ |
|
|
| with gr.Blocks(css="style.css") as demo: |
| gr.HTML("<h1><center>Multilingual Stable Diffusion 🧨</center></h1>") |
| gr.Markdown(description) |
| with gr.Group(): |
| with gr.Row(): |
| prompt = gr.Textbox(label='Enter your prompt', scale=8) |
| submit = gr.Button(scale=1, variant='primary') |
| img = gr.Image(label='Generated Image') |
|
|
| prompt.submit(fn=generate_image, |
| inputs=[prompt], |
| outputs=img, |
| ) |
| submit.click(fn=generate_image, |
| inputs=[prompt], |
| outputs=img, |
| ) |
| |
| demo.queue().launch() |
