Spaces:
Running on T4
Running on T4
Commit ·
331be04
1
Parent(s): 3ed8a24
cuda
Browse files
main.py
CHANGED
|
@@ -11,7 +11,8 @@ import requests
|
|
| 11 |
app = Flask(__name__)
|
| 12 |
CORS(app)
|
| 13 |
|
| 14 |
-
|
|
|
|
| 15 |
|
| 16 |
# Global model setup
|
| 17 |
checkpoint = "sam_vit_l_0b3195.pth"
|
|
|
|
| 11 |
app = Flask(__name__)
|
| 12 |
CORS(app)
|
| 13 |
|
| 14 |
+
cudaOrNah = "cuda" if torch.cuda.is_available() else "cpu"
|
| 15 |
+
print(cudaOrNah)
|
| 16 |
|
| 17 |
# Global model setup
|
| 18 |
checkpoint = "sam_vit_l_0b3195.pth"
|