fffiloni commited on
Commit
d7be7dc
·
verified ·
1 Parent(s): ba29991

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -45,6 +45,17 @@ def generate(input_video):
45
 
46
  try:
47
  # Run the inference command
 
 
 
 
 
 
 
 
 
 
 
48
  subprocess.run(
49
  [
50
  "python", "infer_4d.py", "big",
@@ -58,6 +69,7 @@ def generate(input_video):
58
  )
59
 
60
  output_videos = glob(os.path.join(f"{workdir}", "*.mp4"))
 
61
  return output_videos[0]
62
 
63
  except subprocess.CalledProcessError as e:
 
45
 
46
  try:
47
  # Run the inference command
48
+ subprocess.run(
49
+ [
50
+ "python", "infer_4d.py", "big",
51
+ "--workspace", f"{workdir}",
52
+ "--resume", f"{recon_model}",
53
+ "--num_frames", f"1",
54
+ "--test_path", f"{test_path}",
55
+ ],
56
+ check=True
57
+ )
58
+
59
  subprocess.run(
60
  [
61
  "python", "infer_4d.py", "big",
 
69
  )
70
 
71
  output_videos = glob(os.path.join(f"{workdir}", "*.mp4"))
72
+ print(output_videos)
73
  return output_videos[0]
74
 
75
  except subprocess.CalledProcessError as e: