Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -248,6 +248,7 @@ def special_process_image_yolo(risk_level, image_path, point1, point2, threshold
|
|
248 |
# YOLOv8モデルをロードし、指定デバイスに移動
|
249 |
model = YOLO(model_path).to(device)
|
250 |
print("モデルが正常にロードされ、デバイスに移動しました。")
|
|
|
251 |
|
252 |
# タイムスタンプを作成
|
253 |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|
|
|
248 |
# YOLOv8モデルをロードし、指定デバイスに移動
|
249 |
model = YOLO(model_path).to(device)
|
250 |
print("モデルが正常にロードされ、デバイスに移動しました。")
|
251 |
+
print(model.names)
|
252 |
|
253 |
# タイムスタンプを作成
|
254 |
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
|