Update app.py

app.py

@@ -41,6 +41,7 @@ batch_size = -1
 width = int(os.getenv("IMAGE_WIDTH", "512"))
 height = int(os.getenv("IMAGE_HEIGHT", "512"))
 num_images = int(os.getenv("NUM_IMAGES", "1"))
+guidance_scale = float(os.getenv("GUIDANCE_SCALE", "1.0"))
 
 class CustomOVModelVaeDecoder(OVModelVaeDecoder):
     def __init__(
@@ -78,7 +79,6 @@ def save_images(image_array, profile: gr.OAuthProfile | None, metadata: dict):
 def generate(
     prompt: str,
     seed: int = 0,
-    guidance_scale: float = 1.0,
     num_inference_steps: int = 4,
     randomize_seed: bool = False,
     progress = gr.Progress(track_tqdm=True),
@@ -88,6 +88,7 @@ def generate(
     global width
     global height
     global num_images
+    global guidance_scale
 
     seed = randomize_seed_fn(seed, randomize_seed)
     np.random.seed(seed)
@@ -172,7 +173,6 @@ with gr.Blocks(css="style.css") as demo:
         inputs=[
             prompt,
             seed,
-            1.0,
             num_inference_steps,
             randomize_seed
         ],