Upload multit2i.py

multit2i.py

@@ -163,12 +163,12 @@ def infer(prompt: str, model_name: str, recom_prompt: bool, progress=gr.Progress
     caption = model_name.split("/")[-1]
     try:
         model = load_model(model_name)
-        if not model: return (None, None)
+        if not model: return (Image.new("RGB", (64, 64), (0, 0, 0)), None)
         image_path = model(prompt + rprompt + seed)
         image = Image.open(image_path).convert('RGB')
     except Exception as e:
         print(e)
-        return (None, None)
+        return (Image.new("RGB", (64, 64), (0, 0, 0)), None)
     return (image, caption)