up 1

app.py

@@ -19,7 +19,8 @@ def image_to_byte_array(image: Image) -> bytes:
 def predict(input_img1,
             input_img2,
             positive_prompt,
-            negative_prompt
+            negative_prompt,
+            num_of_images
             ):
 
     print("predict")
@@ -32,14 +33,14 @@ def predict(input_img1,
     canvas_mask = np.array(input_img2)
     mask = get_mask(canvas_mask)
 
-    print(input_img1, mask,positive_prompt,negative_prompt)
+    print(input_img1, mask, positive_prompt, negative_prompt)
 
     retList=[]
-    for x in range(10):
-        result_image = make_inpainting(positive_prompt="empty room",
+    for x in range(num_of_images):
+        result_image = make_inpainting(positive_prompt=positive_prompt,
                                    image=input_img1,
                                    mask_image=mask,
-                                   negative_prompt="",
+                                   negative_prompt=negative_prompt,
                                    )
         retList.append(result_image)
 
@@ -49,7 +50,10 @@ def predict(input_img1,
 app = gr.Interface(
     predict,
     inputs=[gr.Image(label="img", sources=['upload'], type="pil"),
-            gr.Image(label="mask", 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")
             ],
     outputs= [
         gr.Image(label="resp0"),
@@ -68,8 +72,7 @@ app = gr.Interface(
 app.launch(share=True)
 
 #
-#  gr.Textbox(label="positive_prompt"),
-#  gr.Textbox(label="negative_prompt")
+
 
 # gr.Interface(
 #     test1,