Update app.py

app.py

@@ -68,7 +68,17 @@ def send_it1(inputs, model_choice):
     with open(prompt_path, 'w') as f:
         f.write(inputs[0])
 
-    Image.fromarray(output1).save(image_path)
+    # Check the type of output1 before saving
+    if isinstance(output1, np.ndarray):  # If it's a numpy array
+        Image.fromarray(np.uint8(output1)).save(image_path)
+    elif isinstance(output1, Image.Image):  # If it's already a PIL Image
+        output1.save(image_path)
+    elif isinstance(output1, str):  # If it's a string (this should not happen in ideal conditions)
+        print(f"Warning: output1 is a string. Cannot save as image. Value: {output1}")
+    else:
+        print(f"Warning: Unexpected type {type(output1)} for output1.")
+        
+    #Image.fromarray(output1).save(image_path)
 
     saved_output.update(list_saved_prompts_and_images())