up 6

app.py

@@ -29,8 +29,7 @@ def predict(input_img1,
     # input_img1 = Image.fromarray(input_img1)
     # input_img2 = Image.fromarray(input_img2)
 
-    config.WIDTH = resolution
-    config.HEIGHT = resolution
+    config.setResoluton(resolution)
 
     input_img1 = input_img1.resize((config.WIDTH, config.HEIGHT))
     input_img2 = input_img2.resize((config.WIDTH, config.HEIGHT))
@@ -59,10 +58,10 @@ app = gr.Interface(
     predict,
     inputs=[gr.Image(label="img", sources=['upload'], type="pil"),
             gr.Image(label="mask", sources=['upload'], type="pil"),
-            gr.Textbox(label="positive_prompt"),
-            gr.Textbox(label="negative_prompt"),
-            gr.Number(label="num_of_images"),
-            gr.Number(label="resolution")
+            gr.Textbox(label="positive_prompt",value="empty room"),
+            gr.Textbox(label="negative_prompt",value=""),
+            gr.Number(label="num_of_images",value=2),
+            gr.Number(label="resolution",value=512)
             ],
     outputs= [
         gr.Image(label="resp0"),

config.py

@@ -7,6 +7,10 @@ HEIGHT = 512
 WIDTH = 512
 #
 
+def setResoluton(res):
+    global HEIGHT, WIDTH
+    HEIGHT = res
+    WIDTH = res
 
 def to_rgb(color: str) -> tuple:
     """Convert hex color to rgb.