Spaces:

2nzi
/

PnLCalib

Sleeping

2nzi commited on Jun 19

Commit

17a5b18

1 Parent(s): 9a5e1bd

CPU edit api.py

Files changed (1) hide show

api.py CHANGED Viewed

@@ -40,7 +40,7 @@ app.add_middleware(
 # Paramètres par défaut pour l'inférence
 WEIGHTS_KP = "models/SV_FT_TSWC_kp"
 WEIGHTS_LINE = "models/SV_FT_TSWC_lines"
-DEVICE = "cuda:0"
 KP_THRESHOLD = 0.15
 LINE_THRESHOLD = 0.15
 PNL_REFINE = True

 # Paramètres par défaut pour l'inférence
 WEIGHTS_KP = "models/SV_FT_TSWC_kp"
 WEIGHTS_LINE = "models/SV_FT_TSWC_lines"
+DEVICE = "cpu"
 KP_THRESHOLD = 0.15
 LINE_THRESHOLD = 0.15
 PNL_REFINE = True