Update app.py (#4)

- Update app.py (fd550927bbd7bd242664d828e74ba41a29117c34)


Co-authored-by: Roopal Garg <roopalgarg@users.noreply.huggingface.co>

app.py

@@ -12,13 +12,14 @@ except:
     pass
 
 auth_token = os.environ.get('HF_TOKEN', None)
-if not auth_token:
-    raise ValueError("could not authenticate the user.")
 
-iiw_400 = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="IIW-400")
-docci_test = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="DOCCI_Test")
-locnar_eval = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="LocNar_Eval")
-cm_3600 = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="CM_3600")
+try:
+    iiw_400 = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="IIW-400")
+    docci_test = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="DOCCI_Test")
+    locnar_eval = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="LocNar_Eval")
+    cm_3600 = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="CM_3600")
+except Exception as e:
+    logging.error("could you fetch the datasets with error: %s", e)
 
 _SELECTOR_TO_DATASET = {
     "IIW-400": iiw_400,
@@ -36,7 +37,7 @@ def display_iiw_data_with_slider_change(dataset_type, index):
         image_url_key = "image/thumbnail_url"
         image_key = "image"
 
-    logging.warning(f"SELECTION: {dataset_type} : {dataset_split}: {index}")
+    logging.debug(f"SELECTION: {dataset_type} : {dataset_split}: {index}")
     data = _SELECTOR_TO_DATASET[dataset_type][dataset_split][index]
     image_html = f'<img src="{data[image_url_key]}" style="width:100%; max-width:800px; height:auto;">'
     image_key_html = f"<p style='font-size: 10px'>Image Key: {data[image_key]}</p>"
@@ -78,7 +79,7 @@ def max_index(dataset_type):
     if dataset_type == "LocNar_Eval":
         dataset_split = "validation"
 
-    logging.warning(f"SELECTION: {dataset_type} : {dataset_split}")
+    logging.debug(f"SELECTION: {dataset_type} : {dataset_split}")
     dataset_instance =_SELECTOR_TO_DATASET[dataset_type][dataset_split]
     return len(dataset_instance)