Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -227,7 +227,7 @@ def special_process_image_yolo(risk_level, image_path, point1, point2, threshold
|
|
227 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
228 |
print(device)
|
229 |
# YOLOv8モデルをロードし、GPUに移動
|
230 |
-
model = YOLO('
|
231 |
model.to(device) # モデルをGPUに移動
|
232 |
|
233 |
|
|
|
227 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
228 |
print(device)
|
229 |
# YOLOv8モデルをロードし、GPUに移動
|
230 |
+
model = YOLO('./1026.pt') # モデルのパスを指定
|
231 |
model.to(device) # モデルをGPUに移動
|
232 |
|
233 |
|