.

main.py

@@ -142,7 +142,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
 
 # Initialize the model at the start of your application
 #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
-model = initialize_model("pssearch_bsz64_acc1_lr8e5_512_leftclick_histpos_512_384.yaml", "yuntian-deng/computer-model")
+model = initialize_model("standard_challenging_context32_nocond_all.yaml", "yuntian-deng/computer-model")
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.to(device)