test 1024

app.py

@@ -17,7 +17,10 @@ def image_to_byte_array(image: Image) -> bytes:
     return imgByteArr
 
 def predict(input_img1,
-            input_img2):
+            input_img2,
+            positive_prompt,
+            negative_prompt
+            ):
 
     print("predict")
     input_img1 = Image.fromarray(input_img1)
@@ -31,22 +34,35 @@ def predict(input_img1,
 
     print(input_img1, mask)
 
-    result_image = make_inpainting(positive_prompt='an empty room',
+    result_image = make_inpainting(positive_prompt=positive_prompt,
                                    image=input_img1,
                                    mask_image=mask,
-                                   negative_prompt="",
+                                   negative_prompt=negative_prompt,
                                    )
 
     return result_image
 
-gradio_app = gr.Interface(
+def test1(param1):
+    return  "here !"
+
+gr.Interface(
     predict,
     inputs=[gr.Image(label="img", sources=['upload', 'webcam'], type="numpy"),
-            gr.Image(label="mask", sources=['upload', 'webcam'], type="numpy")
+            gr.Image(label="mask", sources=['upload', 'webcam'], type="numpy"),
+            gr.Textbox(label="positive_prompt"),
+            gr.Textbox(label="negative_prompt")
             ],
     outputs= gr.Image(label="resp"),
     title="rem fur 1",
-)
+).launch(share=True)
+#
+#
+# gr.Interface(
+#     test1,
+#     inputs=[gr.Textbox(label="param1")],
+#     outputs= gr.Textbox(label="result"),
+#     title="rem fur 1",
+# ).launch(share=True)
+
 
-gradio_app.launch(share=True)
 

config.py

@@ -1,8 +1,11 @@
 """File with configs"""
 from palette import COLOR_MAPPING_, COLOR_MAPPING
 
-HEIGHT = 512
-WIDTH = 512
+HEIGHT = 1024
+WIDTH = 1024
+# HEIGHT = 512
+# WIDTH = 512
+
 
 def to_rgb(color: str) -> tuple:
     """Convert hex color to rgb.