Update app_local.py

app_local.py

@@ -29,6 +29,7 @@ REWRITER_MODEL = "Qwen/Qwen1.5-4B-Chat"  # Upgraded to 4B for better JSON handli
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
+LOC = os.getenv("QIE")
 
 # Quantization configuration
 bnb_config = BitsAndBytesConfig(
@@ -213,7 +214,7 @@ def polish_prompt(original_prompt: str) -> str:
         
 # Load main image editing pipeline
 pipe = QwenImageEditPipeline.from_pretrained(
-    "Qwen/Qwen-Image-Edit",
+    LOC,
     torch_dtype=dtype
 ).to(device)