Update app.py
Browse files
app.py
CHANGED
|
@@ -105,12 +105,12 @@ terminators = [
|
|
| 105 |
]
|
| 106 |
|
| 107 |
|
| 108 |
-
if torch.cuda.is_available():
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
else:
|
| 112 |
-
|
| 113 |
-
|
| 114 |
|
| 115 |
model = model.to(device)
|
| 116 |
|
|
|
|
| 105 |
]
|
| 106 |
|
| 107 |
|
| 108 |
+
# if torch.cuda.is_available():
|
| 109 |
+
# device = torch.device("cuda")
|
| 110 |
+
# print(f"Using GPU: {torch.cuda.get_device_name(device)}")
|
| 111 |
+
# else:
|
| 112 |
+
# device = torch.device("cpu")
|
| 113 |
+
# print("Using CPU")
|
| 114 |
|
| 115 |
model = model.to(device)
|
| 116 |
|