Update app_2.py

app_2.py

@@ -1155,6 +1155,7 @@ with block:
                 with gr.Row():
                     extracted_objects = gr.Image(type="numpy", label="Extracted Foreground", height=480)
                     extracted_fg = gr.Image(type="numpy", label="Extracted Foreground", height=480)
+                    angles_fg = gr.Image(type="pil", label="Converted Foreground", height=480, visible=False)
 
                 with gr.Row():
                     run_button = gr.Button("Generate alternative angles")
@@ -1232,8 +1233,20 @@ with block:
         example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
         example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
 
-        angles_fg = Image.fromarray(extracted_fg, 'RGB')
+
+        def convert_to_pil(image):
+            try:
+                image = image.astype(np.uint8)
+                return image
+            except Exception as e:
+                print(f"Error converting Numpy image to PIL :{e})
+            return image
+                      
         run_button.click(
+            fn=convert_to_pil, 
+            inputs=extracted_fg, 
+            outputs=angles_fg)
+            .then(
             fn=infer,
             inputs=[
                 text_prompt,