fix interface

app.py

@@ -38,19 +38,31 @@ def infer(prompt, negative_prompt, image):
         prompt,
         cond_input,
         generator=generator,
-        num_images_per_prompt=1,
+        num_images_per_prompt=4,
         num_inference_steps=20
             )
 
     return output
 
-    # you need to pass inputs and outputs according to inference function
-title = "Categorical Conditioning Controlnet for One-Shot Image Stylization."
-description = "This is a demo on ControlNet which generates images based on the style of the conditioning input." 
-# you need to pass your examples according to your inputs
-# each inner list is one example, each element in the list corresponding to a component in the `inputs`.
-examples = [["1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck", "low quality", "wikipe_cond_1.png"]]
-gr.Interface(fn = infer, inputs = ["text", "text", gr.Image(type="pil") ], outputs = "image",
-                    title = title, description = description, examples = examples, theme='gradio/soft').launch()
-
+gr.Interface(
+    infer,
+    inputs=[
+        gr.Textbox(
+            label="Enter prompt",
+            max_lines=1,
+            placeholder="1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck",
+        ),
+        gr.Textbox(
+            label="Enter negative prompt",
+            max_lines=1,
+            placeholder="low quality",
+        ),
+        gr.Image(type="pil"),
+    ],
+    outputs=gr.Gallery().style(grid=[2], height="auto"),
+    title="Generate controlled outputs with Categorical Conditioning on Waifu Diffusion 1.5 beta 2.",
+    description="This Space uses image examples as style conditioning.",
+    examples=[["wikipe_cond_1.png", "1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck"]],
+    allow_flagging=False,
+).launch(enable_queue=True)