Update app.py

app.py

@@ -21,10 +21,10 @@ ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-p
 config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
 
 device = "cuda:0"
-#model = load_model_from_config(config, ckpt, device=device, verbose=False)
-#model = model.to(device).half()
+model = load_model_from_config(config, ckpt, device=device, verbose=False)
+model = model.to(device).half()
 
-#clip_model, preprocess = clip.load("ViT-L/14", device=device)
+clip_model, preprocess = clip.load("ViT-L/14", device=device)
 
 n_inputs = 5