Hugging Face's logo

Spaces:
Michal-Raszkowski
/
style-transfer-adain
Sleeping

App Files Community
1
Michal-Raszkowski commited on
Commit
27a58b0
·
verified ·
1 Parent(s): 38b4e58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  from src.lightning_module import StyleTransferModule
7
 
8
 
9
- MODEL_URL = "https://huggingface.co/Michal-Raszkowski/adain-style-transfer/resolve/main/style-transfer-best.ckpt?download=true"
10
  CHECKPOINT_PATH = "model.ckpt"
11
 
12
  def download_model_if_missing():
 
6
  from src.lightning_module import StyleTransferModule
7
 
8
 
9
+ MODEL_URL = "https://huggingface.co/Michal-Raszkowski/adain-style-transfer/resolve/main/style-transfer-best-v2.ckpt?download=true"
10
  CHECKPOINT_PATH = "model.ckpt"
11
 
12
  def download_model_if_missing():