Update app.py

app.py

@@ -30,12 +30,11 @@ handler.register(sa_args)
 def style_aligned_multidiff(ref_style_prompt, img_generation_prompt):
     view_batch_size = 25  # adjust according to VRAM size
     reference_latent = torch.randn(1, 4, 64, 64,)
-    for target_prompt in target_prompts:
-        images = pipeline_calls.panorama_call(pipeline, 
-                                              [ref_style_prompt, img_generation_prompt], 
-                                              reference_latent=reference_latent, 
-                                              view_batch_size=view_batch_size)
-    
+    images = pipeline_calls.panorama_call(pipeline,
+                                          [ref_style_prompt, img_generation_prompt],
+                                          reference_latent=reference_latent,
+                                          view_batch_size=view_batch_size)
+
     return images, gr.Image(value=images[0], visible=True)
 
 
@@ -44,34 +43,49 @@ with gr.Blocks() as demo:
       with gr.Column(variant='panel'):
         ref_style_prompt = gr.Textbox(
           label='Reference style prompt',
-          info="Enter a Prompt to generate the reference image", 
-          placeholder='a beautiful papercut art design'
+          info='Enter a Prompt to generate the reference image',
+          placeholder='A poster in a papercut art style.'
         )
         ref_style_image = gr.Image(visible=False, label='Reference style image')
-    
+
       with gr.Column(variant='panel'):
         img_generation_prompt = gr.Textbox(
           label='MultiDiffusion Prompt',
-          info="Enter a Prompt to generate panaromic images using Style-aligned combined with MultiDiffusion", 
+          info='Enter a Prompt to generate panaromic images using Style-aligned combined with MultiDiffusion',
+          placeholder= 'A village in a papercut art style.'
           )
-        
-        
-    btn = gr.Button("Style-aligned MultiDiffusion - Generate", size='sm')
-
-    gallery = gr.Gallery(label="Style-Aligned ControlNet - Generated images", 
-                           elem_id="gallery",
-                           columns=5, 
-                           rows=1, 
-                           object_fit="contain", 
-                           height="auto",
-                           allow_preview=True, 
+
+
+    btn = gr.Button('Style-aligned MultiDiffusion - Generate', size='sm')
+
+    gallery = gr.Gallery(label='Style-Aligned ControlNet - Generated images',
+                           elem_id='gallery',
+                           columns=5,
+                           rows=1,
+                           object_fit='contain',
+                           height='auto',
+                           allow_preview=True,
                            preview=True,
                           )
-      
-    btn.click(fn=style_aligned_multidiff, 
-              inputs=[ref_style_prompt, img_generation_prompt], 
-              outputs=gallery, 
-              api_name="style_aligned_multidiffusion")
 
+    btn.click(fn=style_aligned_multidiff,
+              inputs=[ref_style_prompt, img_generation_prompt],
+              outputs=[gallery, ref_style_image],
+              api_name='style_aligned_multidiffusion')
+
+
+    gr.Examples(
+      examples=[
+        ['A poster in a papercut art style.', 'A village in a papercut art style.'],
+        ['A poster in a papercut art style.', 'Futuristic cityscape in a papercut art style.'],
+        ['A poster in a papercut art style.', 'A jungle in a papercut art style.'],
+        ['A poster in a flat design style.', 'Girrafes in a flat design style.'],
+        ['A poster in a flat design style.', 'Houses in a flat design style.'],
+        ['A poster in a flat design style.', 'Mountains in a flat design style.'],
+      ],
+      inputs=[ref_style_prompt, img_generation_prompt],
+      outputs=[gallery, ref_style_image],
+      fn=style_aligned_multidiff,
+      )
 
-demo.launch()
+demo.launch()