added collage

app.py

@@ -29,7 +29,7 @@ from diffusers import StableDiffusionPipeline
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-model = None 
+# model = None 
 model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)
 
 from typing import Mapping
@@ -65,6 +65,8 @@ if (SPACE_ID := os.getenv('SPACE_ID')) is not None:
 else:
     DESCRIPTION = DESCRIPTION.replace("either", "")
 
+DESCRIPTION += "<img src='https://raw.githubusercontent.com/BKHMSI/Word-As-Image-Ar/main/collage.gif' alt='Example of Outputs'/>"
+
 
 warnings.filterwarnings("ignore")
 
@@ -312,9 +314,9 @@ with gr.Blocks() as demo:
             run = gr.Button('Generate')
 
         with gr.Column():
-            result0 = gr.Image(type="filepath", label="Initial Word").style(height=333)
-            result1 = gr.Image(type="filepath", label="Optimization Process").style(height=333)
-            result2 = gr.Image(type="filepath", label="Final Result",visible=False).style(height=333)
+            result0 = gr.Image(type="filepath", label="Initial Word").style(height=170)
+            result1 = gr.Image(type="filepath", label="Optimization Process").style(height=300)
+            result2 = gr.Image(type="filepath", label="Final Result",visible=False).style(height=100)
             
         
     with gr.Row():