upd

app.py

@@ -13,6 +13,11 @@ model = hub.Module(name='ernie_vilg')
 def get_ernie_vilg(text_prompts, style): 
   style = style.split('-')[0]
   results = model.generate_image(text_prompts=text_prompts, style=style, visualization=False)
+  for CPU monitoring
+  # Testing the psutil library for both CPU and RAM performance details
+  print(f"ERNIE CPU percent is: {psutil.cpu_percent()}")
+  print(f"ERNIE virtual memory is : {psutil.virtual_memory().percent}")
+
   return results[0]
   
 sd_inf = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion", use_auth_token=HF_TOKEN)
@@ -33,6 +38,11 @@ def get_chinese_translation(text):  #in_language_first, in_language_second,
   output = output[0]['translation_text']
   print(f"output is:{output}")
   
+  # for CPU monitoring
+  # Testing the psutil library for both CPU and RAM performance details
+  print(f"CPU percent is: {psutil.cpu_percent()}")
+  print(f"virtual memory is : {psutil.virtual_memory().percent}")
+
   return output 
   
 #Block inference not working for stable diffusion   
@@ -41,11 +51,6 @@ def get_sd(translated_txt, samples, steps, scale, seed):
   print(f"translated_txt is : {translated_txt}")
   sd_img_gallery = sd_inf(translated_txt, samples, steps, scale, seed, fn_index=1)[0] 
   
-  #for CPU monitoring
-  # Testing the psutil library for both CPU and RAM performance details
-  print(f"CPU percent is: {psutil.cpu_percent()}")
-  print(f"virtual memory is : {psutil.virtual_memory().percent}")
-    
   return sd_img_gallery
 
 demo = gr.Blocks()