Мясников Филипп Сергеевич commited on
Commit
258a840
1 Parent(s): 90c67b0
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -61,7 +61,7 @@ def projection(img, name, device='cuda'):
61
  return w_plus[0]
62
 
63
 
64
- def inference(img, model):
65
  img.save('out.jpg')
66
  aligned_face = align_face('out.jpg')
67
 
@@ -71,8 +71,8 @@ def inference(img, model):
71
  imageio.imwrite('filename.jpeg', npimage)
72
  return 'filename.jpeg'
73
 
74
- title = "JoJoGAN"
75
- description = "Gradio Demo for JoJoGAN: One Shot Face Stylization. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
76
 
77
  gr.Interface(inference,
78
  [gr.inputs.Image(type="pil")],
 
61
  return w_plus[0]
62
 
63
 
64
+ def inference(img):
65
  img.save('out.jpg')
66
  aligned_face = align_face('out.jpg')
67
 
 
71
  imageio.imwrite('filename.jpeg', npimage)
72
  return 'filename.jpeg'
73
 
74
+ title = "PetBreeder v1.1"
75
+ description = "Gradio Demo for PetBreeder."
76
 
77
  gr.Interface(inference,
78
  [gr.inputs.Image(type="pil")],