Upload modutils.py

modutils.py

@@ -5,6 +5,7 @@ import os
 import re
 from pathlib import Path
 from PIL import Image
+import numpy as np
 import shutil
 import requests
 from requests.adapters import HTTPAdapter
@@ -330,9 +331,8 @@ def save_gallery_history(images, files, history_gallery, history_files, progress
 
 
 def save_image_history(image, gallery, files, model_name: str, progress=gr.Progress(track_tqdm=True)):
-    import numpy as np
     basename = f"{model_name.split('/')[-1]}_{datetime.now(FILENAME_TIMEZONE).strftime('%Y%m%d_%H%M%S')}"
-    if image is None or (not isinstance(image, str) and not isinstance(image, Image.Image)  and not isinstance(image, np.array) and not isinstance(image, tuple)): return gr.update(), gr.update()
+    if image is None or not isinstance(image, (str, Image.Image, np.array, tuple)): return gr.update(), gr.update()
     if not gallery: gallery = []
     if not files: files = []
     filename = f"{basename}.png"