Upload folder using huggingface_hub

app.py

@@ -2,12 +2,21 @@ import gradio as gr
 import os
 import numpy as np
 
-def no_cpu_warning(video_file, dataset, lora, size):
+global current_video_index
+current_video_index = "v"
+
+def load_cached_video(video_file, dataset, lora, size):
     output = "./cached_examples/"
     output += "peft" if lora else "audiocraft"
     output += "_" + dataset + "_" + size + "_" + video_file[-7:-4]
-    chosen_video = np.random.choice(['v', 'v1', 'v2'])
-    output += "_" + chosen_video + ".mp4"
+    output += "_" + globals()["current_video_index"] + ".mp4"
+    # roll trough the video indexes
+    if globals()["current_video_index"] == "v":
+        globals()["current_video_index"] = "v1"
+    elif globals()["current_video_index"] == "v1":
+        globals()["current_video_index"] = "v2"
+    else: 
+        globals()["current_video_index"] = "v"
 
     if not os.path.exists(output):
         print(output)
@@ -18,8 +27,8 @@ def no_cpu_warning(video_file, dataset, lora, size):
     return output
 
 with gr.Blocks() as demo:
-    gr.Markdown('<span style="color:yellow"> **WARNING**: Multiple videos examples have to be downloaded to use this UI. Please be patient, as the videos meight take a moment to download.</span>')
-    gr.Interface(fn=no_cpu_warning,
+    gr.Markdown('<span style="color:yellow"> **WARNING**: Multiple video examples have to be downloaded first to use this UI properly. Please be patient, as the videos meight take a moment to download.</span>')
+    gr.Interface(fn=load_cached_video,
                          inputs=[
                                  gr.Video(value="./videos/n_5.mp4",
                                           label="Original Video",