Upload folder using huggingface_hub

app.py

@@ -1,4 +1,5 @@
 import time
+import spaces
 import gradio as gr
 import torch
 import diffusers
@@ -36,7 +37,7 @@ with gr.Blocks() as demo:
         else:
             print("Left Image is merged!")
 
-
+    @spaces.GPU
     def generate(prompt, seed, steps, height_width, negative_prompt, guidance_scale, method):
 
         pipe.enable_xformers_memory_efficient_attention()
@@ -112,12 +113,10 @@ with gr.Blocks() as demo:
 
         which_image(final_img, width=height_width)
 
-
-        result = f"Baseline image: {end_time_base-start_time_base:.2f} sec | {'ToDo' if method == 'todo' else 'ToMe'} image: {end_time_merge-start_time_merge:.2f} sec"
+        result = f"Baseline image: {end_time_base-start_time_base:.2f} sec  |  {'ToDo' if method == 'todo' else 'ToMe'} image: {end_time_merge-start_time_merge:.2f} sec"
 
         return final_img, result
 
-
     gen.click(generate, inputs=[prompt, seed, steps, height_width, negative_prompt,
                                 guidance_scale, method], outputs=[output_image, result])