saimemrekanat commited on
Commit
a205957
·
1 Parent(s): 744cc3b
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -250,7 +250,7 @@ def main():
250
  gr.Markdown(FOOTER)
251
  loadmodel_button.click(fn=model.load_model,
252
  inputs=[style_type],
253
- outputs=[exstyle, load_info])
254
 
255
 
256
  style_type.change(fn=update_slider,
@@ -259,20 +259,20 @@ def main():
259
 
260
  preprocess_image_button.click(fn=model.detect_and_align_image,
261
  inputs=[input_image, top, bottom, left, right],
262
- outputs=[aligned_face, instyle, input_info])
263
  preprocess_video0_button.click(fn=model.detect_and_align_video,
264
  inputs=[input_video, top, bottom, left, right],
265
- outputs=[aligned_face, instyle, input_info])
266
  preprocess_video1_button.click(fn=model.detect_and_align_full_video,
267
  inputs=[input_video, top, bottom, left, right],
268
- outputs=[aligned_video, instyle, input_info])
269
 
270
  toonify_button.click(fn=model.image_toonify,
271
  inputs=[aligned_face, instyle, exstyle, style_degree, style_type],
272
- outputs=[result_face, output_info])
273
  vtoonify_button.click(fn=model.video_tooniy,
274
  inputs=[aligned_video, instyle, exstyle, style_degree, style_type],
275
- outputs=[result_video, output_info], api_name="video")
276
 
277
  example_images.click(fn=set_example_image,
278
  inputs=example_images,
 
250
  gr.Markdown(FOOTER)
251
  loadmodel_button.click(fn=model.load_model,
252
  inputs=[style_type],
253
+ outputs=[exstyle, load_info], api_name="load-model")
254
 
255
 
256
  style_type.change(fn=update_slider,
 
259
 
260
  preprocess_image_button.click(fn=model.detect_and_align_image,
261
  inputs=[input_image, top, bottom, left, right],
262
+ outputs=[aligned_face, instyle, input_info], api_name="scale-image")
263
  preprocess_video0_button.click(fn=model.detect_and_align_video,
264
  inputs=[input_video, top, bottom, left, right],
265
+ outputs=[aligned_face, instyle, input_info], api_name="scale-first-frame")
266
  preprocess_video1_button.click(fn=model.detect_and_align_full_video,
267
  inputs=[input_video, top, bottom, left, right],
268
+ outputs=[aligned_video, instyle, input_info], api_name="scale-video")
269
 
270
  toonify_button.click(fn=model.image_toonify,
271
  inputs=[aligned_face, instyle, exstyle, style_degree, style_type],
272
+ outputs=[result_face, output_info], api_name="toonify-image")
273
  vtoonify_button.click(fn=model.video_tooniy,
274
  inputs=[aligned_video, instyle, exstyle, style_degree, style_type],
275
+ outputs=[result_video, output_info], api_name="toonify-video")
276
 
277
  example_images.click(fn=set_example_image,
278
  inputs=example_images,