loading model bug fix

app.py

@@ -4,22 +4,17 @@ import spaces
 from PIL import Image
 
 import numpy as np
+import torch
 import torch.nn.functional as F
 
-from utils.pipeline_utils import *
+from utils.pipeline_utils import load_pipeline
 from utils import get_args
 
 from main import run
 
-pipeline = None
+pipeline = load_pipeline(fp16=False, cache_dir=None)
 
 
-def get_pipeline():
-    global pipeline
-    if pipeline is None:
-        pipeline = load_pipeline(fp16=False, cache_dir=None)
-    return pipeline
-
 
 def process_masks(masks):
     # masks: list of file paths
@@ -52,8 +47,6 @@ def main_pipeline(
         dift_correction: bool = True,
 ):
     args = get_args()
-    pipeline = get_pipeline()
-
     args.alpha = alpha
     args.beta = beta
     args.w1 = w1
@@ -70,7 +63,6 @@ def main_pipeline(
 
 DESCRIPTION = """# Cora 🖼️🐱🦅
         ## Fast & Controllable Image Editing
-
         ### 🛠️ Quick start  
         1. **Upload** or drag-and-drop the image you’d like to edit.  
         2. **Source prompt** – describe what’s in the original image.  
@@ -78,9 +70,7 @@ DESCRIPTION = """# Cora 🖼️🐱🦅
         4. Adjust the parameters as needed.  
         5. *(Optional)* Paint a mask to specify the area to edit.  
         6. Click **Edit** and wait a few seconds for the output.
-
         ### ⚙️ Parameter cheat-sheet  
-
         | Parameter | What it does | `0` (minimum) | `1` (maximum) |
         |-----------|--------------|---------------|---------------|
         | **alpha** | Appearance transfer control | preserve source appearance | target prompt affects appearance |
@@ -88,7 +78,6 @@ DESCRIPTION = """# Cora 🖼️🐱🦅
         | **w**     | Prompt strength    | subtle tweaks | strong changes |
         | **Seed**  | Fixes randomness for reproducibility | – | – |
         | **Apply correspondence correction** | Uses correspondence-aware latent fix | – | – |
-
         ### 📜 Tips  
         - To replicate **TurboEdit**, set **alpha = 1**, **beta = 1**, and turn **off**  *Apply correspondence correction*.  
         - To test reconstruction quality of the inversion, use identical source & target prompts with **alpha = 1**, **beta = 1**, and **w = 1**.