Update app.py

app.py

@@ -2,9 +2,13 @@ import gradio as gr
 from optimum.intel.openvino import OVStableDiffusionPipeline
 from diffusers.training_utils import set_seed
 from diffusers import DDPMScheduler, StableDiffusionPipeline
+import gc
+
+import subprocess
 
 import time
 
+
 def create_pipeline(name):
     if name == "valhalla/sd-pokemon-model":
         scheduler = DDPMScheduler(beta_start=0.00085, beta_end=0.012,
@@ -34,6 +38,9 @@ pipes = {
 for v in pipes.values():
     pipe = create_pipeline(v)
     del pipe
+    gc.collect()
+
+print((subprocess.check_output("lscpu", shell=True).strip()).decode())
 
 def generate(prompt, option, seed):
     pipe = create_pipeline(pipes[option])