fix

app.py

@@ -1,14 +1,20 @@
 import numpy as np
 import gradio as gr
 import paddlehub as hub
+import time
 
 
 model = hub.Module(directory='ernie_vilg')
 
 def inference(text_prompts, style):
+  enter_module_time = time.time()
+  print('enter module time ', time.ctime(enter_module_time))
   try:
     results = model.generate_image(
         text_prompts=text_prompts, style=style, visualization=False)
+    exist_module_time = time.time()
+    print('Exist module time ',time.ctime(exist_module_time))
+    print('module exectution time ', exist_module_time-enter_module_time)
     return 'Success', results[:6]
   except Exception as e:
     error_text = str(e)

ernie_vilg/module.py

@@ -76,6 +76,8 @@ class ErnieVilG:
         if not os.path.exists(output_dir):
           os.makedirs(output_dir, exist_ok=True)
         token = self.token
+        task_create_start_time = time.time()
+        print('task_create_start_time', time.ctime(task_create_start_time))
         create_url = 'https://wenxin.baidu.com/younger/portal/api/rest/1.0/ernievilg/v1/txt2img?from=paddlehub'
         get_url = 'https://wenxin.baidu.com/younger/portal/api/rest/1.0/ernievilg/v1/getImg?from=paddlehub'
         if isinstance(text_prompts, str):
@@ -108,8 +110,12 @@ class ErnieVilG:
           else:
             print(res['msg'])
             raise RuntimeError(res['msg'])
+        task_create_end_time = time.time()
+        print('task_create_end_time', time.ctime(task_create_end_time))
+        print('task create time:', task_create_end_time-task_create_start_time)
         
         start_time = time.time()
+        print('task poll start time', time.ctime(start_time))
         process_bar = tqdm(total=100, unit='%')
         results = {}
         first_iter = True
@@ -165,9 +171,11 @@ class ErnieVilG:
           for taskid in has_done:
             taskids.remove(taskid)
         end_time = time.time() #debug code
-        print('request task time', end_time-start_time) #debug code
+        print('task poll end time', time.ctime(end_time)) #debug code
+        print('poll time: ',  end_time-start_time)
         start_save_time = time.time() #debug code
         print('Saving Images...')
+        print('download_start_time', time.ctime(start_save_time)) #debug code
         result_images = []
         for text, data in results.items():
           for idx, imgdata in enumerate(data['imgUrls']):
@@ -186,6 +194,7 @@ class ErnieVilG:
               break
         print('Done')
         end_save_time = time.time() #debug code
+        print('download_end_time', time.ctime(end_save_time)) #debug code
         print('Download and decode images time', end_save_time-start_save_time)
         return result_images