Spaces:
Runtime error
Runtime error
Commit
·
85bd4b9
1
Parent(s):
20a1b21
Update main.py
Browse files
main.py
CHANGED
@@ -53,7 +53,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]])
|
|
53 |
|
54 |
# Initialize the model at the start of your application
|
55 |
#model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
|
56 |
-
model = initialize_model("
|
57 |
|
58 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
59 |
model = model.to(device)
|
|
|
53 |
|
54 |
# Initialize the model at the start of your application
|
55 |
#model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
|
56 |
+
model = initialize_model("2e5_debug_gpt_firstframe_posmap.yaml", "yuntian-deng/computer-model")
|
57 |
|
58 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
59 |
model = model.to(device)
|