Update app.py

app.py

@@ -4,6 +4,8 @@ import sys
 from pathlib import Path
 from PIL import Image
 import re
+from PIL import Image
+import numpy as np
 
 # Coder: Create directories if they don't exist
 if not os.path.exists('saved_prompts'):
@@ -137,12 +139,18 @@ with gr.Blocks(css=css) as myface:
         with gr.Column(scale=100):
             magic1=gr.Textbox(lines=4)
             run=gr.Button("Generate Image")
+            
     with gr.Row():
         with gr.Column(scale=100):
             model_name1 = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model, interactive=True)
+
     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:
+                Image.fromarray(np.array(output1, dtype=np.uint8)).save(image_path)
                 
     with gr.Row():
         with gr.Column(scale=50):