Upload folder using huggingface_hub

app.py

@@ -42,6 +42,7 @@ for style_name, style_img_path in style_options.items():
 
 @spaces.GPU(duration=10)
 def run(content_image, style_name, style_strength, output_quality, progress=gr.Progress(track_tqdm=True)):
+    yield None
     img_size = 1024 if output_quality else 512
     content_img, original_size = preprocess_img(content_image, img_size)
     content_img = content_img.to(device)
@@ -66,7 +67,7 @@ def run(content_image, style_name, style_strength, output_quality, progress=gr.P
     et = time.time()
     print('TIME TAKEN:', et-st)
     
-    return postprocess_img(generated_img, original_size)
+    yield postprocess_img(generated_img, original_size)
 
 
 def set_slider(value):