api

app.py

@@ -39,7 +39,7 @@ DESCRIPTION = '''
 FOOTER = '<div align=center><img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.laobi.icu/badge?page_id=williamyang1991/VToonify" /></div>'
 
 ARTICLE = r"""
-If VToonify is helpful, please help to ⭐ the <a href='https://github.com/williamyang1991/VToonify' target='_blank'>Github Repo</a>. Thanks! 
+If VToonify is helpful, please help to ⭐ the <a href='https://github.com/williamyang1991/VToonify' target='_blank'>Github Repo</a>. Thanks!
 [![GitHub Stars](https://img.shields.io/github/stars/williamyang1991/VToonify?style=social)](https://github.com/williamyang1991/VToonify)
 ---
 📝 **Citation**
@@ -60,7 +60,7 @@ If our work is useful for your research, please consider citing:
 ```
 
 📋 **License**
-This project is licensed under <a rel="license" href="https://github.com/williamyang1991/VToonify/blob/main/LICENSE.md">S-Lab License 1.0</a>. 
+This project is licensed under <a rel="license" href="https://github.com/williamyang1991/VToonify/blob/main/LICENSE.md">S-Lab License 1.0</a>.
 Redistribution and use for non-commercial purposes should follow this license.
 
 📧 **Contact**
@@ -77,22 +77,23 @@ def set_example_image(example: list) -> dict:
     return gr.Image.update(value=example[0])
 
 def set_example_video(example: list) -> dict:
-    return gr.Video.update(value=example[0]), 
-   
+    return gr.Video.update(value=example[0]),
+
 sample_video = ['./vtoonify/data/529_2.mp4','./vtoonify/data/7154235.mp4','./vtoonify/data/651.mp4','./vtoonify/data/908.mp4']
 sample_vid = gr.Video(label='Video file')  #for displaying the example
-example_videos = gr.components.Dataset(components=[sample_vid], samples=[[path] for path in sample_video], type='values', label='Video Examples')     
+example_videos = gr.components.Dataset(components=[sample_vid], samples=[[path] for path in sample_video], type='values', label='Video Examples')
 
 def main():
+    btn.click(add, [num1, num2], output, api_name="addition")
     args = parse_args()
     args.device = 'cuda' if torch.cuda.is_available() else 'cpu'
     print('*** Now using %s.'%(args.device))
     model = Model(device=args.device)
-    
+
     with gr.Blocks(theme=args.theme, css='style.css') as demo:
-        
+
         gr.Markdown(DESCRIPTION)
-    
+
         with gr.Box():
             gr.Markdown('''## Step 1(Select Style)
     - Select **Style Type**.
@@ -102,16 +103,16 @@ def main():
     ''')
             with gr.Row():
                 with gr.Column():
-                    gr.Markdown('''Select Style Type''')  
+                    gr.Markdown('''Select Style Type''')
                     with gr.Row():
                         style_type = gr.Radio(label='Style Type',
                                               choices=['cartoon1','cartoon1-d','cartoon2-d','cartoon3-d',
                                                        'cartoon4','cartoon4-d','cartoon5-d','comic1-d',
                                                        'comic2-d','arcane1','arcane1-d','arcane2', 'arcane2-d',
                                                        'caricature1','caricature2','pixar','pixar-d',
-                                                       'illustration1-d', 'illustration2-d', 'illustration3-d', 'illustration4-d', 'illustration5-d', 
+                                                       'illustration1-d', 'illustration2-d', 'illustration3-d', 'illustration4-d', 'illustration5-d',
                                                       ]
-                                             )   
+                                             )
                         exstyle = gr.Variable()
                     with gr.Row():
                         loadmodel_button = gr.Button('Load Model')
@@ -119,7 +120,7 @@ def main():
                         load_info = gr.Textbox(label='Process Information', interactive=False, value='No model loaded.')
                 with gr.Column():
                     gr.Markdown('''Reference Styles
-    ![example](https://raw.githubusercontent.com/williamyang1991/tmpfile/master/vtoonify/style.jpg)''')   
+    ![example](https://raw.githubusercontent.com/williamyang1991/tmpfile/master/vtoonify/style.jpg)''')
 
 
         with gr.Box():
@@ -161,19 +162,19 @@ def main():
                                             256,
                                             value=200,
                                             step=8,
-                                            label='right')     
+                                            label='right')
                 with gr.Box():
                     with gr.Column():
-                        gr.Markdown('''Input''')                
+                        gr.Markdown('''Input''')
                         with gr.Row():
                             input_image = gr.Image(label='Input Image',
                                                            type='filepath')
                         with gr.Row():
-                            preprocess_image_button = gr.Button('Rescale Image') 
+                            preprocess_image_button = gr.Button('Rescale Image')
                         with gr.Row():
                             input_video = gr.Video(label='Input Video',
                                                    mirror_webcam=False,
-                                                          type='filepath')  
+                                                          type='filepath')
                         with gr.Row():
                             preprocess_video0_button = gr.Button('Rescale First Frame')
                             preprocess_video1_button = gr.Button('Rescale Video')
@@ -191,7 +192,7 @@ def main():
                         with gr.Row():
                             aligned_video = gr.Video(label='Rescaled Video',
                                             type='mp4',
-                                            interactive=False)  
+                                            interactive=False)
             with gr.Row():
                 with gr.Column():
                     paths = ['./vtoonify/data/pexels-andrea-piacquadio-733872.jpg','./vtoonify/data/i5R8hbZFDdc.jpg','./vtoonify/data/yRpe13BHdKw.jpg','./vtoonify/data/ILip77SbmOE.jpg','./vtoonify/data/077436.jpg','./vtoonify/data/081680.jpg']
@@ -200,15 +201,15 @@ def main():
                                                label='Image Examples')
                 with gr.Column():
                     #example_videos = gr.Dataset(components=[input_video], samples=[['./vtoonify/data/529.mp4']], type='values')
-                    #to render video example on mouse hover/click        
+                    #to render video example on mouse hover/click
                     example_videos.render()
                     #to load sample video into input_video upon clicking on it
-                    def load_examples(video):  
+                    def load_examples(video):
                         #print("****** inside load_example() ******")
                         #print("in_video is : ", video[0])
                         return video[0]
 
-                    example_videos.click(load_examples, example_videos, input_video) 
+                    example_videos.click(load_examples, example_videos, input_video)
 
         with gr.Box():
             gr.Markdown('''## Step 3 (Generate Style Transferred Image/Video)''')
@@ -224,10 +225,10 @@ def main():
                                              1,
                                              value=0.5,
                                              step=0.05,
-                                             label='Style Degree')  
+                                             label='Style Degree')
                 with gr.Column():
                     gr.Markdown('''![example](https://raw.githubusercontent.com/williamyang1991/tmpfile/master/vtoonify/degree.jpg)
-                        ''')  
+                        ''')
             with gr.Row():
                 output_info = gr.Textbox(label='Process Information', interactive=False, value='n.a.')
             with gr.Row():
@@ -242,10 +243,10 @@ def main():
                     with gr.Row():
                         result_video = gr.Video(label='Result Video',
                                             type='mp4',
-                                            interactive=False)    
+                                            interactive=False)
                     with gr.Row():
                         vtoonify_button = gr.Button('VToonify!')
-        
+
         gr.Markdown(ARTICLE)
         gr.Markdown(FOOTER)
 
@@ -279,7 +280,7 @@ def main():
         example_images.click(fn=set_example_image,
                                  inputs=example_images,
                                  outputs=example_images.components)
-        
+
     demo.launch(
         enable_queue=args.enable_queue,
         server_port=args.port,