Kingoteam commited on
Commit
3f4a0b9
·
verified ·
1 Parent(s): a6ec3f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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((128, 128))
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=128,
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)