Spaces:
Running
on
Zero
Running
on
Zero
cuda
Browse files
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 |
-
|
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 |
|