Update app.py

app.py

@@ -110,14 +110,15 @@ def run_lora(prompt, cfg_scale, steps, lora_repo, lora_name, randomize_seed, see
     final_image = generate_image(prompt, steps, seed, cfg_scale, width, height, lora_scale, progress)
     
     if upload_to_r2:
-        url = upload_image_to_r2(final_image, account_id, access_key, secret_key, bucket)
-        result = {"status": "success", "url": url}
+        with calculateDuration("upload r2"):
+            url = upload_image_to_r2(final_image, account_id, access_key, secret_key, bucket)
+            result = {"status": "success", "url": url}
     else:
         result = {"status": "success", "message": "Image generated but not uploaded"}
     
     progress(100, "Completed!")
 
-    yield final_image, seed, gr.update(value=progress_bar, visible=False),  json.dumps(result)
+    yield final_image, seed, json.dumps(result)
 
 
 css="""
@@ -160,7 +161,6 @@ with gr.Blocks(css=css) as demo:
         
 
         with gr.Column():
-            progress_bar = gr.Markdown(elem_id="progress",visible=False)
             result = gr.Image(label="Result", show_label=False)
             json_text = gr.Text()
 
@@ -168,7 +168,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn = run_lora,
         inputs = [prompt, cfg_scale, steps, lora_repo, lora_name, randomize_seed, seed, width, height, lora_scale, upload_to_r2, account_id, access_key, secret_key, bucket],
-        outputs=[result, seed, progress_bar, json_text]
+        outputs=[result, seed, json_text]
     )
 
 demo.queue().launch()