Update app.py

app.py

@@ -4,6 +4,7 @@ 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__)
 
@@ -14,10 +15,8 @@ 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(f'models/{model_name}')
+            models_load[model_name] = DiffusionPipeline.from_pretrained(models[model_name])
         except Exception as error:
             print(f"Error loading model {model_name}: {error}")
             models_load[model_name] = None