Layout changes

app.py

@@ -28,7 +28,7 @@ class Demo:
 
     def layout(self):
         with gr.Row():
-            with gr.Column() as training_column:
+            with gr.Column(scale=1) as training_column:
                 self.prompt_input = gr.Text(
                     placeholder="Enter prompt...",
                     label="Prompt to Erase",
@@ -66,15 +66,24 @@ class Demo:
                 )
                 
                 
-            with gr.Column() as inference_column:
+            with gr.Column(scale=2) as inference_column:
 
                 with gr.Row():
 
-                    self.prompt_input_infr = gr.Text(
-                        placeholder="Enter prompt...",
-                        label="Prompt",
-                        info="Prompt to generate"
-                    )
+                    with gr.Column(scale=4):
+
+                        self.prompt_input_infr = gr.Text(
+                            placeholder="Enter prompt...",
+                            label="Prompt",
+                            info="Prompt to generate"
+                        )
+
+                    with gr.Column(scale=1):
+
+                        self.seed_infr = gr.Number(
+                            label="Seed",
+                            value=42
+                        )
 
                 with gr.Row():
 
@@ -95,6 +104,7 @@ class Demo:
                     )
                     self.infr_button.click(self.inference, inputs = [
                             self.prompt_input_infr,
+                            self.seed_infr
                         ],
                         outputs=[
                             self.image_new,
@@ -156,7 +166,7 @@ class Demo:
         self.training = False
 
 
-    def inference(self, prompt):
+    def inference(self, prompt, seed, pbar = gr.Progress(track_tqdm=True)):
 
         if self.generating:
             return [None, None]
@@ -165,6 +175,8 @@ class Demo:
 
         self.diffuser.unet.load_state_dict(self.model_orig_sd)
 
+        self.diffuser._seed = seed
+
         images = self.diffuser(
             prompt,
             n_steps=50,