Update app.py

app.py

@@ -200,13 +200,11 @@ def process_and_generate(google_api_key, stability_api_key, prompt, style, steps
      State("stability-api-key", "value"),
      State("prompt", "value"),
      State("style", "value"),
-     State("cfg-scale", "value"),
      State("steps", "value"),
      State("aspect-ratio", "value")],
     prevent_initial_call=True
 )
-
-def update_output(n_clicks, google_api_key, stability_api_key, prompt, style, cfg_scale, steps, aspect_ratio):
+def update_output(n_clicks, google_api_key, stability_api_key, prompt, style, steps, aspect_ratio):
     if n_clicks is None:
         raise PreventUpdate
     
@@ -218,7 +216,7 @@ def update_output(n_clicks, google_api_key, stability_api_key, prompt, style, cf
         status["message"] = message
     
     def run_process():
-        image_bytes, enhanced_prompt = process_and_generate(google_api_key, stability_api_key, prompt, style, cfg_scale, steps, aspect_ratio, set_status)
+        image_bytes, enhanced_prompt = process_and_generate(google_api_key, stability_api_key, prompt, style, steps, aspect_ratio, set_status)
         if image_bytes:
             encoded_image = base64.b64encode(image_bytes).decode('ascii')
             return f"data:image/jpeg;base64,{encoded_image}", f"Enhanced Prompt: {enhanced_prompt}", status["message"], False