Spaces:

SkalskiP
/

better-florence-2

Running on Zero

SkalskiP commited on Jul 9

Commit

a3dae10

•

1 Parent(s): fc7652c

cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -56,8 +56,8 @@ EXAMPLES = [
     ["microsoft/Florence-2-large-ft", OCR_WITH_REGION_TASK_NAME, "https://media.roboflow.com/inference/license_plate_1.jpg", None],
 ]
-DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-# DEVICE = "cuda"
 MODELS, PROCESSORS = load_models(DEVICE)

     ["microsoft/Florence-2-large-ft", OCR_WITH_REGION_TASK_NAME, "https://media.roboflow.com/inference/license_plate_1.jpg", None],
 ]
+# DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+DEVICE = "cuda"
 MODELS, PROCESSORS = load_models(DEVICE)