Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 import gradio as gr
+import uuid
+import shutil
 from openvideo import push_file_to_hf
 try:
     import ml4co_kit
@@ -9,27 +11,36 @@ except:
 from ml4co_kit import CVRPPyVRPSolver
 from ml4co_kit import CVRPDataGenerator
 
-FILEPATH = "data/cvrp/uniform/cvrp50_uniform.txt"
+FILEPATH = "data/cvrp/uniform/cvrp100_uniform.txt"
 
 def handle(
-    hf_token: str, 
-    filename: str, 
+    hf_token: str
 ):
-    solver = CVRPPyVRPSolver(time_limit=10)
+    solver = CVRPPyVRPSolver(time_limit=60)
     gen = CVRPDataGenerator(
         num_threads=8,
+        nodes_num=100,
+        min_capacity=50,
+        max_capacity=50,
         solver=solver,
-        train_samples_num=89600,
+        train_samples_num=256,
         val_samples_num=0,
-        test_samples_num=0,
-    )
-    gen.generate()
-    push_file_to_hf(
-        hf_token=hf_token,
-        hf_repo_id="ML4CO/ML4VRP",
-        file_path=FILEPATH,
-        path_in_repo=filename
+        test_samples_num=0
     )
+    cur_iter = 0
+    max_iter = 1000
+    
+    while(cur_iter < max_iter):
+        gen.generate()
+        filename = uuid.uuid4().hex[:9] + ".txt"
+        push_file_to_hf(
+            hf_token=hf_token,
+            hf_repo_id="ML4CO/ML4VRPDataset",
+            file_path=FILEPATH,
+            path_in_repo=filename
+        )
+        shutil.rmtree("data/cvrp")
+        cur_iter = cur_iter + 1
     
     
 with gr.Blocks() as demo:
@@ -39,7 +50,6 @@ with gr.Blocks() as demo:
         '''
     )
     hf_token = gr.Textbox(label="HuggingFace Token")
-    filename = gr.Textbox(label="txt name")
 
     with gr.Row():
         button = gr.Button("Submit", variant="primary")
@@ -47,7 +57,7 @@ with gr.Blocks() as demo:
 
     button.click(
         handle, 
-        [hf_token, filename], 
+        [hf_token], 
         outputs=None
     )