Update app.py

app.py

@@ -18,7 +18,6 @@ from safetensors.torch import load_file
 dtype = torch.bfloat16
 from huggingface_hub import snapshot_download
 model_path  = snapshot_download(repo_id="nyanko7/flux-dev-de-distill")
-print("model_list ",os.listdir(model_path))
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # ---------------- Encoders ----------------
 
@@ -633,7 +632,7 @@ class EmptyInitWrapper(torch.overrides.TorchFunctionMode):
 with EmptyInitWrapper():
     model = Flux().to(dtype=torch.bfloat16, device="cuda")
     
-    sd = load_file("./consolidated_s6700.safetensors")
+    sd = load_file(f"{model_path}/consolidated_s6700.safetensors")
     sd = {k.replace("model.", ""): v for k, v in sd.items()}
     result = model.load_state_dict(sd)