Update app.py

app.py

@@ -147,11 +147,14 @@ with gr.Blocks(css=css) as myface:
     with gr.Row():
         with gr.Column(style="width=800px"):
             output1=gr.Image(label=(f"{current_model}"))
-            if isinstance(output1, str):
-                print(f"Warning: output1 is a string, not an array. Value: {output1}")
-            else:
+            # Check the type before attempting to save the image
+            if isinstance(output1, Image.Image):  # Check if it's a PIL Image object
+                output1.save(image_path)
+            elif isinstance(output1, np.ndarray):  # Check if it's a NumPy array
                 Image.fromarray(np.array(output1, dtype=np.uint8)).save(image_path)
-                
+            else:
+                print(f"Warning: Unexpected type {type(output1)} for output1.")
+                            
     with gr.Row():
         with gr.Column(scale=50):
             input_text=gr.Textbox(label="Prompt Idea",lines=2)