Update app.py

app.py

@@ -23,7 +23,7 @@ print("downloaded!")
 from src.flux.xflux_pipeline import XFluxPipeline
 
 @spaces.GPU(duration=200)
-def process_image(lora_path, lora_name, number_of_images, image, prompt, steps, use_lora, use_controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg):
+def process_image(lora_path, lora_name, image, prompt, steps, use_lora, use_controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg):
     def run_xflux_pipeline(
         prompt, image, repo_id, name, device,
         model_type, width, height, timestep_to_start_cfg, num_steps, true_gs, guidance,
@@ -92,7 +92,7 @@ def process_image(lora_path, lora_name, number_of_images, image, prompt, steps,
         
         # Laço para gerar imagens
         images = []
-        for _ in range(args.num_images_per_prompt):
+        for _ in range(1):
             seed = random.randint(0, 2147483647)
             result = xflux_pipeline(
                 prompt=args.prompt,
@@ -130,7 +130,7 @@ def process_image(lora_path, lora_name, number_of_images, image, prompt, steps,
           height=1024,
           timestep_to_start_cfg=cfg,
           num_steps=steps,
-          num_images_per_prompt=number_of_images,
+          num_images_per_prompt=1,
           use_lora=use_lora,
           lora_repo_id=lora_path,
           lora_name=lora_name,
@@ -168,7 +168,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=2, elem_classes="app"):
             output = gr.Gallery(label="Galery output", elem_classes="galery")
 
-    submit_btn.click(process_image, inputs=[lora_path, lora_name, 1, input_image, prompt, steps, use_lora, controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg], outputs=output)
+    submit_btn.click(process_image, inputs=[lora_path, lora_name, input_image, prompt, steps, use_lora, controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg], outputs=output)
 
 if __name__ == '__main__':
     demo.launch(share=True, debug=True)