Update app.py

app.py

@@ -53,7 +53,7 @@ pipe = EllaXLPipeline(pipe,f'{pipeline_path}/pytorch_model.bin')
 # print(f"Optimizing finished successfully after {time.time()-t} secs")
 
 @spaces.GPU(enable_queue=True)
-def infer(prompt,negative_prompt,seed,resolution):
+def infer(prompt,negative_prompt,seed,resolution, steps):
     print(f"""
     —/n
     {prompt}
@@ -73,7 +73,7 @@ def infer(prompt,negative_prompt,seed,resolution):
 
     w,h = resolution.split()
     w,h = int(w),int(h)
-    image = pipe(prompt,num_inference_steps=30, negative_prompt=negative_prompt,generator=generator,width=w,height=h).images[0]
+    image = pipe(prompt,num_inference_steps=steps, negative_prompt=negative_prompt,generator=generator,width=w,height=h).images[0]
     print(f'gen time is {time.time()-t} secs')
     
     # Future
@@ -104,6 +104,7 @@ with gr.Blocks(css=css) as demo:
                 prompt_in = gr.Textbox(label="Prompt", value="A smiling man with wavy brown hair and a trimmed beard")
                 resolution = gr.Dropdown(value=resolutions[0], show_label=True, label="Resolution", choices=resolutions)
                 seed = gr.Textbox(label="Seed", value=-1)
+                steps = gr.Textbox(label="Steps", value=50)
                 negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
                 submit_btn = gr.Button("Generate")
         result = gr.Image(label="BRIA-2.4 Result")
@@ -131,7 +132,8 @@ with gr.Blocks(css=css) as demo:
             prompt_in,
             negative_prompt,
             seed,
-            resolution
+            resolution,
+            steps
         ],
         outputs = [
             result