Update app.py

app.py

@@ -4,7 +4,6 @@ from flask import Flask, request, send_file
 from io import BytesIO
 from PIL import Image, ImageChops
 from diffusers import DiffusionPipeline
-from all_models import models  # Import the models variable from all_models
 
 app = Flask(__name__)
 
@@ -15,8 +14,10 @@ def load_model(model_name):
     global models_load
     if model_name not in models_load:
         try:
+            # ディレクトリ内のファイルリストを出力
+            print("Available models:", os.listdir('models'))
             # モデルのロード
-            models_load[model_name] = DiffusionPipeline.from_pretrained(models[model_name])
+            models_load[model_name] = DiffusionPipeline.from_pretrained(f'models/{model_name}')
         except Exception as error:
             print(f"Error loading model {model_name}: {error}")
             models_load[model_name] = None