akhaliq HF staff commited on
Commit
548fea0
1 Parent(s): aa20d61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -122,17 +122,17 @@ modelv4 = torch.jit.load(modelarcanev4,map_location='cpu').eval().cpu().half().f
122
  modelv3 = torch.jit.load(modelarcanev3,map_location='cpu').eval().cpu().half().float()
123
  modelv2 = torch.jit.load(modelarcanev2,map_location='cpu').eval().cpu().half().float()
124
 
125
- def version4(img):
126
  im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
127
  res = proc_pil_img(im, modelv4)
128
  return res
129
 
130
- def version3(img):
131
  im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
132
  res = proc_pil_img(im, modelv3)
133
  return res
134
 
135
- def version2(img):
136
  im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
137
  res = proc_pil_img(im, modelv2)
138
  return res
122
  modelv3 = torch.jit.load(modelarcanev3,map_location='cpu').eval().cpu().half().float()
123
  modelv2 = torch.jit.load(modelarcanev2,map_location='cpu').eval().cpu().half().float()
124
 
125
+ def version4(im):
126
  im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
127
  res = proc_pil_img(im, modelv4)
128
  return res
129
 
130
+ def version3(im):
131
  im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
132
  res = proc_pil_img(im, modelv3)
133
  return res
134
 
135
+ def version2(im):
136
  im = scale_by_face_size(im, target_face=256, max_res=1_500_000, max_upscale=1)
137
  res = proc_pil_img(im, modelv2)
138
  return res