update

app.py

@@ -5,11 +5,9 @@ import os
 
 # auth_token = os.environ.get("auth_token")
 auth_token = os.environ.get("HF_TOKEN")
-print("1")
 Visual_Riddles = load_dataset("nitzanguetta/Visual_Riddles", token=auth_token, trust_remote_code=True)['test']
 # print(f"Loaded WHOOPS!, first example:")
 # print(whoops[0])
-print("HI")
 dataset_size = len(Visual_Riddles)
 
 IMAGE = 'Image'
@@ -25,7 +23,6 @@ CATEGORY = "Category"
 DESIGNER = "Designer"
 
 
-print("HI2")
 left_side_columns = [IMAGE]
 right_side_columns = [x for x in Visual_Riddles.features.keys() if x not in left_side_columns]
 right_side_columns.remove('Image file name')
@@ -36,11 +33,9 @@ emoji_to_label = {IMAGE: '🎨, 🧑‍🎨, 💻', ANSWER: '💡, 🤔, 🧑‍
 # batch_size = 16
 batch_size = 8
 target_size = (1024, 1024)
-print("HI3")
 
 
 def func(index):
-    print("func")
     start_index = index * batch_size
     end_index = start_index + batch_size
     all_examples = [Visual_Riddles[index] for index in list(range(start_index, end_index))]
@@ -52,7 +47,6 @@ def func(index):
 
 
 def get_instance_values(example):
-    print("get_instance_values")
     values = []
     for k in left_side_columns + right_side_columns:
         if k == IMAGE:
@@ -73,7 +67,6 @@ demo = gr.Blocks()
 
 
 def get_col(example):
-    print("get_col")
     instance_values = get_instance_values(example)
     with gr.Column():
         inputs_left = []
@@ -100,31 +93,18 @@ def get_col(example):
 
 
 with demo:
-    print("HI4")
     gr.Markdown("# Slide to iterate Visual Riddles")
 
     with gr.Column():
-        print("HI5")
         num_batches = math.ceil(dataset_size / batch_size)
-        print("HI6")
         slider = gr.Slider(minimum=0, maximum=num_batches, step=1, label=f'Page (out of {num_batches})')
-        print("HI7")
         with gr.Row():
-            print("HI8")
             index = slider.value
             start_index = 0 * batch_size
             end_index = start_index + batch_size
-            print("HI9")
-            print(f"batch_size:{batch_size}")
-            print(f"start_index:{start_index}")
-            print(f"end_index:{end_index}")
-            print(f"Visual_Riddles:{Visual_Riddles}")
-            print(f"Visual_Riddles:{Visual_Riddles[0]}")
             all_examples = [Visual_Riddles[index] for index in list(range(start_index, end_index))]
-            print("HI10")
             all_inputs_left_right = []
             for example_idx, example in enumerate(all_examples):
-                print("HI11")
                 inputs_left, text_inputs_right = get_col(example)
                 inputs_left_right = inputs_left + text_inputs_right
                 all_inputs_left_right += inputs_left_right