SkalskiP commited on
Commit
a3dae10
1 Parent(s): fc7652c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -56,8 +56,8 @@ EXAMPLES = [
56
  ["microsoft/Florence-2-large-ft", OCR_WITH_REGION_TASK_NAME, "https://media.roboflow.com/inference/license_plate_1.jpg", None],
57
  ]
58
 
59
- DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
60
- # DEVICE = "cuda"
61
  MODELS, PROCESSORS = load_models(DEVICE)
62
 
63
 
 
56
  ["microsoft/Florence-2-large-ft", OCR_WITH_REGION_TASK_NAME, "https://media.roboflow.com/inference/license_plate_1.jpg", None],
57
  ]
58
 
59
+ # DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
60
+ DEVICE = "cuda"
61
  MODELS, PROCESSORS = load_models(DEVICE)
62
 
63