Update predict.py
Browse files- predict.py +1 -0
predict.py
CHANGED
|
@@ -27,6 +27,7 @@ class Predictor(BasePredictor):
|
|
| 27 |
) -> Path:
|
| 28 |
version = model.split(":")[-1]
|
| 29 |
predictor = MyPredictor(model, version)
|
|
|
|
| 30 |
img = read_image(str(image))
|
| 31 |
anime_img = predictor.transform(resize_image(img))[0]
|
| 32 |
out_path = Path(tempfile.mkdtemp()) / "out.png"
|
|
|
|
| 27 |
) -> Path:
|
| 28 |
version = model.split(":")[-1]
|
| 29 |
predictor = MyPredictor(model, version)
|
| 30 |
+
print(f"image path: {str(image)}")
|
| 31 |
img = read_image(str(image))
|
| 32 |
anime_img = predictor.transform(resize_image(img))[0]
|
| 33 |
out_path = Path(tempfile.mkdtemp()) / "out.png"
|