Update app.py

app.py

@@ -22,6 +22,7 @@ def load_model(model_name):
     if model_name not in models_load:
         try:
             m = gr.load(f'models/{model_name}')
+            print(f"Model {model_name} loaded successfully.")
             models_load[model_name] = m
         except Exception as error:
             print(f"Error loading model {model_name}: {error}")
@@ -54,6 +55,9 @@ def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=None,
             # モデル呼び出し
             result = models_load[model_str](*inputs)
             
+            # Debugging result type
+            print(f"Result type: {type(result)}, Result: {result}")
+
             # Check if result is an image or a file path
             if isinstance(result, str):  # Assuming result might be a file path
                 if os.path.exists(result):
@@ -65,7 +69,7 @@ def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=None,
                 image = result
             else:
                 print("Result is not an image:", type(result))
-                return None, 'Result is not an image'
+                return None, f"Unexpected result type: {type(result)}"
             
             # Check if the image is completely black
             black = Image.new('RGB', image.size, (0, 0, 0))