Update app.py

app.py

@@ -130,10 +130,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     for image in image_generator:
         step_counter+=1
         final_image = image
-        progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
-        yield image, seed, gr.update(value=progress_bar, visible=True)
+        yield image, seed, gr.update(value=progress, visible=True)
         
-    yield final_image, seed, gr.update(value=progress_bar, visible=False) 
+    yield final_image, seed, gr.update(value=progress, visible=False) 
     
 
 run_lora.zerogpu = True
@@ -151,7 +150,6 @@ css = '''
 #progress{height:30px}
 #progress .generating{display:none}
 .progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px}
-.progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
 '''
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     title = gr.HTML(
@@ -185,7 +183,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
             )
             
         with gr.Column(scale=4):
-            progress_bar = gr.Markdown(elem_id="progress",visible=False)
             result = gr.Image(label="Generated Image")
 
     with gr.Row():
@@ -215,7 +212,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
         inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
-        outputs=[result, seed, progress_bar]
+        outputs=[result, seed]
     )
 
 app.queue(default_concurrency_limit=2).launch(show_error=True)