Update app.py

app.py

@@ -1,7 +1,7 @@
 import dataclasses
 import json
 from pathlib import Path
-
+import os
 import gradio as gr
 import torch
 from PIL import Image
@@ -54,6 +54,11 @@ def create_demo(
     device = torch.device(device)
     dtype = torch.float16 if device.type == "cuda" else torch.float32
 
+    # Set Number CPU core is maximum
+    num_cpu_cores = os.cpu_count()  # Adjust based on your system
+    os.environ["OMP_NUM_THREADS"] = str(num_cpu_cores)
+    torch.set_num_threads(num_cpu_cores)
+    
     # Load models with consistent dtype
     vae = AutoencoderKL.from_pretrained(model_name, subfolder="vae", torch_dtype=dtype).to(device)
     tokenizer = CLIPTokenizer.from_pretrained(model_name, subfolder="tokenizer")