Spaces:
Running
Running
Update app.py
Browse files
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("
|
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 |
|