try to get access with secret

README.md

@@ -2,9 +2,8 @@
 title: "RAP: Risk-Aware Prediction"
 emoji: 🚙
 colorFrom: red
-colorTo: grey
+colorTo: gray
 sdk: gradio
-sdk_version: 3.7
 app_file: app.py
 pinned: false
 language: 

scripts/scripts_utils/plotly_interface.py

@@ -348,17 +348,17 @@ def main(load_from=None, cfg_path=None):
     # Define the device to use
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
-    predictor, dataset = load_from_huggingface(device=device)
-    if load_from is not None:
-        cfg = Config.fromfile(cfg_path)
-        predictor = get_predictor(cfg, WaymoDataloaders.unnormalize_trajectory)
-        predictor = load_weights(predictor, torch.load(load_from, map_location="cpu"))
-
-    ui_update_fn = partial(update_figure, predictor, dataset)
-
     # Do the same thing as above but using the gradio blocks API
     with gr.Blocks() as interface:
+        
+        predictor, dataset = load_from_huggingface(device=device)
+        
+        if load_from is not None:
+            cfg = Config.fromfile(cfg_path)
+            predictor = get_predictor(cfg, WaymoDataloaders.unnormalize_trajectory)
+            predictor = load_weights(predictor, torch.load(load_from, map_location="cpu"))
 
+        ui_update_fn = partial(update_figure, predictor, dataset)
         gr.Markdown(
         """
             # Risk-Aware Prediction