Update app_inference.py

app_inference.py

@@ -33,12 +33,16 @@ css = """
 #title h1 {
     font-size: 250%;
 }
-
+    
 .lora-title {
-    color: white;
-    text-align: center;
-    border-radius: 10px;
-    background: linear-gradient(90deg, #1CB5E0 0%, #000851 100%);
+            background-image: linear-gradient(to right, #314755 0%, #26a0da  51%, #314755  100%);
+            text-align: center;        
+            border-radius: 10px;
+            display: block;
+          }
+
+.lora-title h2 {
+    color: white !important;
 }
 
 .gr-image {
@@ -49,13 +53,10 @@ css = """
 }
 
 .res-image {
-    width: 720px;
-    height: 720px;
     object-fit: contain;
     margin: auto;
 }
 
-
 .lora-column {
     display: flex;
     flex-direction: column;
@@ -193,7 +194,7 @@ with gr.Blocks(css=css) as demo:
                     max_lines=1,
                     placeholder='Example: "A [c] in [s] style"'
                 )
-                result = gr.Image(label='Result', elem_classes="res-image")
+                result = gr.Gallery(label='Result', elem_classes="res-image")
                 with gr.Accordion('Other Parameters', open=False, elem_classes="gr-accordion"):
                     content_alpha = gr.Slider(label='Content B-LoRA alpha',
                                               minimum=0,
@@ -220,6 +221,11 @@ with gr.Blocks(css=css) as demo:
                                                maximum=50,
                                                step=0.1,
                                                value=7.5)
+                    num_images_per_prompt = gr.Slider(label='Number of Images per Prompt',
+                                                      minimum=1,
+                                                      maximum=4,
+                                                      step=1,
+                                                      value=4)
                 run_button = gr.Button('Generate')
     demo.load(demo_init, inputs=[],
               outputs=[content_lora_model_id, content_prompt, content_image, style_lora_model_id, style_prompt,
@@ -263,6 +269,7 @@ with gr.Blocks(css=css) as demo:
         seed,
         num_steps,
         guidance_scale,
+        num_images_per_prompt
     ]
     prompt.submit(fn=pipe.run, inputs=inputs, outputs=result)
     run_button.click(fn=pipe.run, inputs=inputs, outputs=result)