Spaces:
Runtime error
Runtime error
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 |
|