Leo8613 commited on
Commit
cbe01ed
·
verified ·
1 Parent(s): 20654d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def preprocess_frame(frame):
58
  return tensor_frame
59
 
60
  def main(video_path):
61
- model = load_model("model.pth")
62
  output_video_path = colorize_video(video_path, model)
63
  return output_video_path
64
 
 
58
  return tensor_frame
59
 
60
  def main(video_path):
61
+ model = load_model("ColorizeVideo_gen.pth")
62
  output_video_path = colorize_video(video_path, model)
63
  return output_video_path
64