UI update

app.py

@@ -5,28 +5,6 @@ from PIL import Image
 
 from model import *
 
-# SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", "0") == "1"
-
-# Constants
-# base = "stabilityai/stable-diffusion-xl-base-1.0"
-# repo = "ByteDance/SDXL-Lightning"
-# checkpoints = {
-#     "1-Step" : ["sdxl_lightning_1step_unet_x0.safetensors", 1],
-#     "2-Step" : ["sdxl_lightning_2step_unet.safetensors", 2],
-#     "4-Step" : ["sdxl_lightning_4step_unet.safetensors", 4],
-#     "8-Step" : ["sdxl_lightning_8step_unet.safetensors", 8],
-# }
-# loaded = None
-
-
-# Ensure model and scheduler are initialized in GPU-enabled function
-# if torch.cuda.is_available():
-#     pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")
-
-    
-# Function 
-# @spaces.GPU(enable_queue=True)
-
 def generate_image(prompt):
     
     return prompt_to_img(prompt)[0]
@@ -35,19 +13,18 @@ def generate_image(prompt):
 
 # Gradio Interface
 description = """
-This demo utilizes the SDXL-Lightning model by ByteDance, which is a lightning-fast text-to-image generative model capable of producing high-quality images in 4 steps.
-As a community effort, this demo was put together by AngryPenguin. Link to model: https://huggingface.co/ByteDance/SDXL-Lightning
+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>Text-to-Image with SDXL-Lightning ⚡</center></h1>")
+    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 (English)', scale=8)
-            ckpt = gr.Dropdown(label='Select inference steps',choices=['1-Step', '2-Step', '4-Step', '8-Step'], value='4-Step', interactive=True)
+            prompt = gr.Textbox(label='Enter your prompt', scale=8)
             submit = gr.Button(scale=1, variant='primary')
-    img = gr.Image(label='SDXL-Lightning Generated Image')
+    img = gr.Image(label='Generated Image')
 
     prompt.submit(fn=generate_image,
                  inputs=[prompt],