JuyeopDang commited on
Commit
01d5b32
ยท
verified ยท
1 Parent(s): 5ab5cab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -1
app.py CHANGED
@@ -13,4 +13,22 @@ else:
13
  torch_dtype = torch.float32
14
 
15
  if __name__ == "__main__":
16
- demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  torch_dtype = torch.float32
14
 
15
  if __name__ == "__main__":
16
+ from huggingface_hub import hf_hub_download
17
+ CONFIG_PATH = 'configs/composite_config.yaml'
18
+
19
+ repo_id = "JuyeopDang/KoFace-Diffusion"
20
+ filename = "composite_epoch2472.pth" # ์˜ˆ: "pytorch_model.pt" ๋˜๋Š” "model.pt"
21
+ vae = VariationalAutoEncoder(CONFIG_PATH)
22
+
23
+ try:
24
+ # ํŒŒ์ผ ๋‹ค์šด๋กœ๋“œ
25
+ # cache_dir์„ ์ง€์ •ํ•˜๋ฉด ๋‹ค์šด๋กœ๋“œ๋œ ํŒŒ์ผ์ด ์ €์žฅ๋  ๊ฒฝ๋กœ๋ฅผ ์ œ์–ดํ•  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค.
26
+ # ๊ธฐ๋ณธ์ ์œผ๋กœ๋Š” ~/.cache/huggingface/hub ์— ์ €์žฅ๋ฉ๋‹ˆ๋‹ค.
27
+ model_path = hf_hub_download(repo_id=repo_id, filename=filename)
28
+ print(f"๋ชจ๋ธ ๊ฐ€์ค‘์น˜ ํŒŒ์ผ์ด ๋‹ค์Œ ๊ฒฝ๋กœ์— ๋‹ค์šด๋กœ๋“œ๋˜์—ˆ์Šต๋‹ˆ๋‹ค: {model_path}")
29
+ except Exception as e:
30
+ print(f"ํŒŒ์ผ ๋‹ค์šด๋กœ๋“œ ๋˜๋Š” ๋ชจ๋ธ ๋กœ๋“œ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: {e}")
31
+
32
+ state_dict = torch.load(model_path, map_location='cuda')
33
+ vae.load_state_dict(state_dict['model_state_dict'])
34
+ print(vae)