akin23 commited on
Commit
e8cc0e9
·
verified ·
1 Parent(s): 8e7f259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -11
app.py CHANGED
@@ -4,32 +4,42 @@ import os
4
  import requests
5
  from datetime import datetime
6
 
7
- # 🔽 Eğer checkpoint klasöründe dosyalar yoksa Hugging Face'ten indir
8
  def download_if_missing(url, local_path):
9
  if not os.path.exists(local_path):
10
  print(f"⬇️ İndiriliyor: {url}")
11
  r = requests.get(url)
 
12
  with open(local_path, "wb") as f:
13
  f.write(r.content)
 
 
14
 
15
- def fetch_model_parts():
 
16
  os.makedirs("checkpoints", exist_ok=True)
17
  base_url = "https://huggingface.co/akin23/SadTalker-Checkpoints/resolve/main"
18
 
19
- parts = [
20
  "facevid2vid_00189-model_aa",
21
  "facevid2vid_00189-model_bb",
22
  "facevid2vid_00189-model_cc",
23
  "facevid2vid_00189-model_dd",
24
- "facevid2vid_00189-model_ee"
 
 
 
 
 
 
25
  ]
26
 
27
- for part in parts:
28
- full_url = f"{base_url}/{part}"
29
- local_path = f"checkpoints/{part}"
30
- download_if_missing(full_url, local_path)
31
 
32
- # 🔧 Model parçalarını birleştirme fonksiyonu
33
  def merge_model_parts():
34
  parts = [
35
  "checkpoints/facevid2vid_00189-model_aa",
@@ -49,8 +59,8 @@ def merge_model_parts():
49
  else:
50
  print("🟡 Birleştirilmiş dosya zaten var.")
51
 
52
- # 📦 Model parçalarını indir ve birleştir
53
- fetch_model_parts()
54
  merge_model_parts()
55
 
56
  # 🔁 Ana video üretim fonksiyonu
 
4
  import requests
5
  from datetime import datetime
6
 
7
+ # 🔽 Dosya yoksa Hugging Face'ten indir
8
  def download_if_missing(url, local_path):
9
  if not os.path.exists(local_path):
10
  print(f"⬇️ İndiriliyor: {url}")
11
  r = requests.get(url)
12
+ r.raise_for_status()
13
  with open(local_path, "wb") as f:
14
  f.write(r.content)
15
+ else:
16
+ print(f"✅ Zaten var: {local_path}")
17
 
18
+ # 📦 Tüm model dosyalarını indir
19
+ def fetch_all_checkpoints():
20
  os.makedirs("checkpoints", exist_ok=True)
21
  base_url = "https://huggingface.co/akin23/SadTalker-Checkpoints/resolve/main"
22
 
23
+ files = [
24
  "facevid2vid_00189-model_aa",
25
  "facevid2vid_00189-model_bb",
26
  "facevid2vid_00189-model_cc",
27
  "facevid2vid_00189-model_dd",
28
+ "facevid2vid_00189-model_ee",
29
+ "epoch_20.pth",
30
+ "mapping_00109-model.pth",
31
+ "mapping_00229-model.pth.tar",
32
+ "auido2exp_00300-model.pth",
33
+ "auido2pose_00140-model.pth",
34
+ "shape_predictor_68_face_landmarks.dat"
35
  ]
36
 
37
+ for filename in files:
38
+ url = f"{base_url}/{filename}"
39
+ local_path = f"checkpoints/{filename}"
40
+ download_if_missing(url, local_path)
41
 
42
+ # 🔧 facevid2vid parçalarını birleştir
43
  def merge_model_parts():
44
  parts = [
45
  "checkpoints/facevid2vid_00189-model_aa",
 
59
  else:
60
  print("🟡 Birleştirilmiş dosya zaten var.")
61
 
62
+ # 🧩 Dosyaları indir ve birleştir
63
+ fetch_all_checkpoints()
64
  merge_model_parts()
65
 
66
  # 🔁 Ana video üretim fonksiyonu