Ahsen Khaliq commited on
Commit
48b9ade
1 Parent(s): 2a66f25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -135,13 +135,13 @@ modelv2 = torch.jit.load('./ArcaneGANv0.2.jit').eval().cuda().half()
135
 
136
  def process(im, version):
137
  if version == 'version 0.4':
138
- im = scale_by_face_size(im, target_face=300, max_res=1_500_000, max_upscale=2)
139
  res = proc_pil_img(im, modelv4)
140
  elif version == 'version 0.3':
141
- im = scale_by_face_size(im, target_face=300, max_res=1_500_000, max_upscale=2)
142
  res = proc_pil_img(im, modelv3)
143
  else:
144
- im = scale_by_face_size(im, target_face=300, max_res=1_500_000, max_upscale=2)
145
  res = proc_pil_img(im, modelv2)
146
  return res
147
 
@@ -151,7 +151,7 @@ article = "<div style='text-align: center;'>ArcaneGan by <a href='https://twitte
151
 
152
  gr.Interface(
153
  process,
154
- [gr.inputs.Image(type="pil", label="Input",shape=(256,256)),gr.inputs.Radio(choices=['version 0.2','version 0.3','version 0.4'], type="value", default='version 0.4', label='version')
155
  ],
156
  gr.outputs.Image(type="pil", label="Output"),
157
  title=title,
135
 
136
  def process(im, version):
137
  if version == 'version 0.4':
138
+ im = scale_by_face_size(im, target_face=300, max_res=1_500_000, max_upscale=1)
139
  res = proc_pil_img(im, modelv4)
140
  elif version == 'version 0.3':
141
+ im = scale_by_face_size(im, target_face=300, max_res=1_500_000, max_upscale=1)
142
  res = proc_pil_img(im, modelv3)
143
  else:
144
+ im = scale_by_face_size(im, target_face=300, max_res=1_500_000, max_upscale=1)
145
  res = proc_pil_img(im, modelv2)
146
  return res
147
 
151
 
152
  gr.Interface(
153
  process,
154
+ [gr.inputs.Image(type="pil", label="Input"),gr.inputs.Radio(choices=['version 0.2','version 0.3','version 0.4'], type="value", default='version 0.4', label='version')
155
  ],
156
  gr.outputs.Image(type="pil", label="Output"),
157
  title=title,