Spaces:
Sleeping
Sleeping
Update interface_connection.py
Browse files- interface_connection.py +2 -2
interface_connection.py
CHANGED
|
@@ -7,10 +7,10 @@ import json
|
|
| 7 |
|
| 8 |
device =torch.device('cpu')
|
| 9 |
model_name_map = {
|
| 10 |
-
"
|
| 11 |
}
|
| 12 |
data_map = {
|
| 13 |
-
"aBN_01": 0, "aBN_02": 1,
|
| 14 |
}
|
| 15 |
|
| 16 |
|
|
|
|
| 7 |
|
| 8 |
device =torch.device('cpu')
|
| 9 |
model_name_map = {
|
| 10 |
+
"wizard_tb 0.3": "demo_model_1"
|
| 11 |
}
|
| 12 |
data_map = {
|
| 13 |
+
"aBN_01": 0, "aBN_02": 1,
|
| 14 |
}
|
| 15 |
|
| 16 |
|