Update app.py
Browse files
app.py
CHANGED
|
@@ -42,15 +42,15 @@ def animegan2_transform(input_img, style_name):
|
|
| 42 |
else:
|
| 43 |
raise ValueError("فرمت تصویر ورودی صحیح نیست!")
|
| 44 |
|
| 45 |
-
#
|
| 46 |
-
input_img = input_img.resize((
|
| 47 |
logger.info("Image resized")
|
| 48 |
|
| 49 |
model = load_model(style_name)
|
| 50 |
face2paint_func = torch.hub.load(
|
| 51 |
"bryandlee/animegan2-pytorch:main",
|
| 52 |
"face2paint",
|
| 53 |
-
size=
|
| 54 |
verbose=False
|
| 55 |
)
|
| 56 |
output_img = face2paint_func(model, input_img)
|
|
|
|
| 42 |
else:
|
| 43 |
raise ValueError("فرمت تصویر ورودی صحیح نیست!")
|
| 44 |
|
| 45 |
+
# افزایش رزولوشن به 760x760
|
| 46 |
+
input_img = input_img.resize((760, 760))
|
| 47 |
logger.info("Image resized")
|
| 48 |
|
| 49 |
model = load_model(style_name)
|
| 50 |
face2paint_func = torch.hub.load(
|
| 51 |
"bryandlee/animegan2-pytorch:main",
|
| 52 |
"face2paint",
|
| 53 |
+
size=760, # افزایش رزولوشن خروجی
|
| 54 |
verbose=False
|
| 55 |
)
|
| 56 |
output_img = face2paint_func(model, input_img)
|