Update app.py

app.py

@@ -110,7 +110,11 @@ if uploaded_file is not None:
             
             with st.spinner("A IA está fazendo a mágica!"):
                 output = process_inpaint(np.array(img_input), np.array(im)) #TODO Put button here
-                img_output = Image.fromarray(output).convert("RGB")
+                #img_output = Image.fromarray(output).convert("RGB")
+                img_input = Image.open(BytesIO(bytes_data))
+                if img_input.mode != "RGBA":
+                    img_input = img_input.convert("RGBA")
+
             
             st.write("AI terminou o trabalho!")
             st.image(img_output)