Spaces:
Runtime error
Runtime error
Commit
·
8a16fb0
1
Parent(s):
5ecddc8
Update main.py
Browse files
main.py
CHANGED
@@ -122,7 +122,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
|
|
122 |
|
123 |
# Initialize the model at the start of your application
|
124 |
#model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
|
125 |
-
model = initialize_model("
|
126 |
|
127 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
128 |
model = model.to(device)
|
|
|
122 |
|
123 |
# Initialize the model at the start of your application
|
124 |
#model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
|
125 |
+
model = initialize_model("pssearch_bsz64_acc1_lr8e5_512_leftclick_histpos_512_384.yaml", "yuntian-deng/computer-model")
|
126 |
|
127 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
128 |
model = model.to(device)
|