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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -3,6 +3,14 @@ import subprocess
3
  import os
4
  from datetime import datetime
5
 
 
 
 
 
 
 
 
 
6
  def generate_video(source_image, driven_audio):
7
  output_dir = "results"
8
  os.makedirs(output_dir, exist_ok=True)
 
3
  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"
16
  os.makedirs(output_dir, exist_ok=True)