Update app.py

app.py

@@ -11,7 +11,7 @@ import collections
 from datasets import load_dataset
 
 dataset_small = load_dataset("pierreguillou/DocLayNet-small")
-dataset_base = load_dataset("pierreguillou/DocLayNet-base")
+#dataset_base = load_dataset("pierreguillou/DocLayNet-base")
 
 id2label = {idx:label for idx,label in enumerate(dataset_small["train"].features["categories"].feature.names)}
 labels = [label for idx, label in id2label.items()]
@@ -243,12 +243,12 @@ with gr.Blocks(title="DocLayNet image viewer", css=".gradio-container") as demo:
     gr.HTML("""
     <div style="font-family:'Times New Roman', 'Serif'; font-size:26pt; font-weight:bold; text-align:center;"><h1>DocLayNet image viewer</h1></div>
     <div style="margin-top: 20px"><p>(01/29/2023) This APP is an image viewer of the DocLayNet dataset.</p></div>
-    <div><p>It uses the datasets <a style="text-decoration: none; border-bottom: #64b5f6 0.125em solid; color: #64b5f6" href="https://huggingface.co/datasets/pierreguillou/DocLayNet-small" target="_blank">DocLayNet small</a> and <a style="text-decoration: none; border-bottom: #64b5f6 0.125em solid; color: #64b5f6" style="text-decoration: none; border-bottom: #64b5f6 0.125em solid; color: #64b5f6" href="https://huggingface.co/datasets/pierreguillou/DocLayNet-base" target="_blank">DocLayNet base</a> (check the corresponding <a href="https://github.com/piegu/language-models/blob/master/DocLayNet_image_viewer_APP.ipynb" target="_blank">notebook</a> to get the full code of this APP).</p></div>
+    <div><p>It uses the dataset <a style="text-decoration: none; border-bottom: #64b5f6 0.125em solid; color: #64b5f6" href="https://huggingface.co/datasets/pierreguillou/DocLayNet-small" target="_blank">DocLayNet small</a> (in the corresponding <a href="https://github.com/piegu/language-models/blob/master/DocLayNet_image_viewer_APP.ipynb" target="_blank">notebook</a>, it is possible to use the <a style="text-decoration: none; border-bottom: #64b5f6 0.125em solid; color: #64b5f6" style="text-decoration: none; border-bottom: #64b5f6 0.125em solid; color: #64b5f6" href="https://huggingface.co/datasets/pierreguillou/DocLayNet-base" target="_blank">DocLayNet base</a> as well the notebook is run in a system setup more powerful than Google Colab).</p></div>
     <div><p>Make your parameters selections and the output will show 2 images of a randomly selected PDF with annotated bounding boxes, one of paragraphs and the other of lines, and a table of texts with their labels.</p></div>
     """)
     with gr.Row():
         with gr.Column():
-            dataset_name_gr = gr.Radio(dataset_names, value="small", label="DocLayNet dataset")
+            dataset_name_gr = gr.Radio(["small"], value="small", label="DocLayNet dataset")
         with gr.Column():
             split_gr = gr.Dropdown(splits, value="all", label="Split")
         with gr.Column():