Update app.py
Browse files
app.py
CHANGED
|
@@ -99,7 +99,8 @@ parsing_model = Parsing(0)
|
|
| 99 |
openpose_model = OpenPose(0)
|
| 100 |
|
| 101 |
# モデルをGPUに転送
|
| 102 |
-
parsing_model.model.to(device)
|
|
|
|
| 103 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
| 104 |
UNet_Encoder.requires_grad_(False)
|
| 105 |
image_encoder.requires_grad_(False)
|
|
|
|
| 99 |
openpose_model = OpenPose(0)
|
| 100 |
|
| 101 |
# モデルをGPUに転送
|
| 102 |
+
# 修正前: parsing_model.model.to(device)
|
| 103 |
+
parsing_model.parsenet.to(device) # 修正後: 正しい属性名を使用
|
| 104 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
| 105 |
UNet_Encoder.requires_grad_(False)
|
| 106 |
image_encoder.requires_grad_(False)
|