fromarray

app.py

@@ -178,7 +178,7 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
                                                         positive_prompt=st.session_state['positive_prompt'],
                                                         negative_prompt=st.session_state['negative_prompt'],
                                                         seed=random.randint(0, 100000) # nosec
-                                                        )[0]
+                                                        )
                 if isinstance(result_image, np.ndarray):
                     result_image = Image.fromarray(result_image)
                 st.session_state['output_image'] = result_image
@@ -252,7 +252,7 @@ def make_editing_canvas(canvas_color, brush, _reset_state, generation_mode, pain
             with st.spinner(text="Generating new images"):
                 print("Making image")
                 result_image = make_inpainting(positive_prompt=st.session_state['positive_prompt'],
-                                                image=image,
+                                                image=Image.fromarray(image),
                                                 mask_image=mask,
                                                 negative_prompt=st.session_state['negative_prompt'],
                                                 )[0]