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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -4
app.py CHANGED
@@ -1,8 +1,34 @@
1
  import gradio as gr
2
  import subprocess
3
  import os
 
4
  from datetime import datetime
5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  # 🔧 Model parçalarını birleştirme fonksiyonu
7
  def merge_model_parts():
8
  parts = [
@@ -14,8 +40,6 @@ def merge_model_parts():
14
  ]
15
  output_file = "checkpoints/facevid2vid_00189-model.pth.tar"
16
 
17
- os.makedirs("checkpoints", exist_ok=True) # <-- klasörü oluştur
18
-
19
  if not os.path.exists(output_file):
20
  with open(output_file, "wb") as out:
21
  for part in parts:
@@ -25,8 +49,8 @@ def merge_model_parts():
25
  else:
26
  print("🟡 Birleştirilmiş dosya zaten var.")
27
 
28
-
29
- # 🔥 Birleştirme işlemini başta yap
30
  merge_model_parts()
31
 
32
  # 🔁 Ana video üretim fonksiyonu
 
1
  import gradio as gr
2
  import subprocess
3
  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 = [
 
40
  ]
41
  output_file = "checkpoints/facevid2vid_00189-model.pth.tar"
42
 
 
 
43
  if not os.path.exists(output_file):
44
  with open(output_file, "wb") as out:
45
  for part in 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