Update app.py

app.py

@@ -5,9 +5,8 @@ from diffusers import DiffusionPipeline
 import torch
 import time
 import psutil
-from huggingface_hub import snapshot_download # New import
+from huggingface_hub import snapshot_download 
 
-# ... (Other imports and variables remain the same)
 
 def load_pipeline(model_id):
     if model_id in PIPELINES:
@@ -21,9 +20,6 @@ def load_pipeline(model_id):
         return pipe
 
 
-# ... (Rest of your code remains the same)
-
-
 # Get the number of physical CPU cores (excluding hyperthreads)
 NUM_CPU_CORES = psutil.cpu_count(logical=True)