Update app.py

app.py

@@ -27,7 +27,7 @@ def load_model(model_name):
             print(f"Error loading model {model_name}: {error}")
             models_load[model_name] = gr.Interface(lambda txt: None, ['text'], ['image'])
 
-def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=0):
+def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=None):
     if model_str not in models_load:
         load_model(model_str)  # モデルがロードされていない場合はロードする
 
@@ -38,8 +38,11 @@ def gen_fn(model_str, prompt, negative_prompt=None, noise=None, cfg_scale=0):
         try:
             if negative_prompt:
                 full_prompt += f' -{negative_prompt}'
-            # Assuming models_load[model_str] accepts cfg_scale as a parameter
-            result = models_load[model_str](full_prompt, cfg_scale=cfg_scale)
+            # Adjust function call based on whether cfg_scale is provided
+            if cfg_scale is not None:
+                result = models_load[model_str](full_prompt, cfg_scale=cfg_scale)
+            else:
+                result = models_load[model_str](full_prompt)
             # 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,10 +68,11 @@ def home():
     model = request.args.get('model', '')
     negative_prompt = request.args.get('Nprompt', None)
     noise = request.args.get('noise', None)
-    cfg_scale = request.args.get('cfg_scale', 0)
+    cfg_scale = request.args.get('cfg_scale', None)
     
     try:
-        cfg_scale = float(cfg_scale)
+        if cfg_scale is not None:
+            cfg_scale = float(cfg_scale)
     except ValueError:
         return 'Invalid "cfg_scale" parameter. It should be a number.', 400