Update app.py

app.py

@@ -100,12 +100,12 @@ def generate(
 
     # Improved options handling
     options = {
-        "prompt": [prompt] * num_images,
-        "negative_prompt": [negative_prompt] * num_images if use_negative_prompt else None,
-        "width": width,
-        "height": height,
-        "guidance_scale": guidance_scale,
-        "num_inference_steps": num_inference_steps,
+        "prompt": [prompt] * int(num_images),
+        "negative_prompt": [negative_prompt] * int(num_images) if use_negative_prompt else None,
+        "width": int(width),
+        "height": int(height),
+        "guidance_scale": float(guidance_scale),
+        "num_inference_steps": int(num_inference_steps),
         "generator": generator,
         "output_type": "pil",
     }
@@ -116,7 +116,7 @@ def generate(
 
     # Generate images potentially in batches
     images = []
-    for i in range(0, num_images, BATCH_SIZE):
+    for i in range(0, int(num_images), BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
         if "negative_prompt" in batch_options:
@@ -252,7 +252,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             num_inference_steps,
             randomize_seed,
             num_images,
-            style_selection,
+            style_selection  # Added style_selection to inputs
         ],
         outputs=[result, seed],
         api_name="run",