.

app.py

@@ -37,8 +37,8 @@ def gradio_ui():
             train_dataset_dropdown = gr.Dropdown(choices=["QCD", "900_03", "900_03+700_07", "700_07", "900_03+700_07+QCD"], label="Training Dataset", value="QCD")
 
         with gr.Row():
-            subdataset_dropdown = gr.Dropdown(choices=[x for x in os.listdir("demo_datasets") if not x.startswith(".")], label="Subdataset")
-            event_idx_dropdown = gr.Dropdown(choices=list(range(20)), label="Event Index")
+            subdataset_dropdown = gr.Dropdown(choices=[x for x in os.listdir("demo_datasets") if not x.startswith(".")], label="Subdataset", value="QCD")
+            event_idx_dropdown = gr.Dropdown(choices=list(range(20)), label="Event Index", value=15)
         prefill_btn = gr.Button("Load Event from Dataset")
 
         particles_text = gr.Textbox(label="Particles CSV (pt eta phi mass charge)", lines=6, interactive=True)

src/model_wrapper_gradio.py

@@ -41,7 +41,7 @@ def inference(loss_str, train_dataset_str, input_text, input_text_quarks):
     args.spatial_part_only = True # LGATr
     args.load_model_weights =  model_path
     args.aug_soft = True # LGATr_GP etc.
-    args.network_config = "src/1models/LGATr/lgatr.py"
+    args.network_config = "src/models/LGATr/lgatr.py"
     args.beta_type = "pt+bc"
     args.embed_as_vectors = False
     args.debug = False
@@ -167,7 +167,7 @@ def inference(loss_str, train_dataset_str, input_text, input_text_quarks):
     for j in range(len(jets_pt)):
         if jets_pt[j] >= 30:
             ax[1].text(jets_eta[j] + 0.1, jets_phi[j] + 0.1,
-                                   "pt=" + str(round(jets_pt[j].item(), 1)), color="gray", fontsize=6, alpha=0.5)
+                                   "pt=" + str(round(jets_pt[j].item(), 1)), color="blue", fontsize=6, alpha=0.5)
             model_jets.append({"pt": jets_pt[j].item(), "eta": jets_eta[j].item(), "phi": jets_phi[j].item()})
 
         if jets_pt[j] >= 100: