akin23 commited on
Commit
a8f5a96
·
verified ·
1 Parent(s): d9a8c1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -6
app.py CHANGED
@@ -4,12 +4,28 @@ import os
4
  from datetime import datetime
5
 
6
  # 👇 Bunu buraya ekle
7
- def merge_checkpoints():
8
- parts = ["facevid2vid_aa", "facevid2vid_ab", "facevid2vid_ac"] # varsa devamı
9
- with open("checkpoints/facevid2vid_00189-model.pth.tar", "wb") as outfile:
10
- for part in parts:
11
- with open(f"checkpoints/{part}", "rb") as infile:
12
- outfile.write(infile.read())
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
 
14
  def generate_video(source_image, driven_audio):
15
  output_dir = "results"
 
4
  from datetime import datetime
5
 
6
  # 👇 Bunu buraya ekle
7
+ import os
8
+
9
+ def merge_model_parts():
10
+ parts = [
11
+ "facevid2vid_00189-model_aa",
12
+ "facevid2vid_00189-model_bb",
13
+ "facevid2vid_00189-model_cc",
14
+ "facevid2vid_00189-model_dd",
15
+ "facevid2vid_00189-model_ee"
16
+ ]
17
+ output_file = "checkpoints/facevid2vid_00189-model.pth.tar"
18
+
19
+ # Sadece yoksa birleştir
20
+ if not os.path.exists(output_file):
21
+ with open(output_file, "wb") as out:
22
+ for part in parts:
23
+ with open(part, "rb") as p:
24
+ out.write(p.read())
25
+ print("✅ Model parçaları birleştirildi.")
26
+ else:
27
+ print("🟡 Birleştirilmiş dosya zaten var.")
28
+
29
 
30
  def generate_video(source_image, driven_audio):
31
  output_dir = "results"