first commit

app.py

@@ -1,9 +1,10 @@
 import random
-
 import gradio as gr
 from datasets import load_dataset
+import os
 
-whoops = load_dataset("nlphuji/whoops")['test']
+auth_token = os.environ.get("auth_token")
+whoops = load_dataset("nlphuji/whoops", use_auth_token=auth_token)['test']
 print(f"Loaded WMTIS, first example:")
 print(whoops[0])
 dataset_size = len(whoops)
@@ -67,7 +68,7 @@ def plot_image(index):
 
 columns_number = 4
 # rows_number = int(dataset_size / columns_number)
-rows_number = 25
+rows_number = 50
 whoops_sample = whoops.shuffle().select(range(0, columns_number * rows_number))
 index = 0