Spaces:
Running
Running
Joshua Lochner
commited on
Commit
•
4d4de75
1
Parent(s):
9f15397
Fix logging messages in predict script
Browse files- src/predict.py +2 -2
src/predict.py
CHANGED
@@ -419,11 +419,11 @@ def main():
|
|
419 |
predictions = predict(video_id, model, tokenizer,
|
420 |
segmentation_args, classifier_args=classifier_args)
|
421 |
except TranscriptError:
|
422 |
-
logger.warning('No transcript available for
|
423 |
continue
|
424 |
video_url = f'https://www.youtube.com/watch?v={video_id}'
|
425 |
if not predictions:
|
426 |
-
logger.info('No predictions found for
|
427 |
continue
|
428 |
|
429 |
# TODO use predict_args.output_as_json
|
|
|
419 |
predictions = predict(video_id, model, tokenizer,
|
420 |
segmentation_args, classifier_args=classifier_args)
|
421 |
except TranscriptError:
|
422 |
+
logger.warning(f'No transcript available for {video_id}')
|
423 |
continue
|
424 |
video_url = f'https://www.youtube.com/watch?v={video_id}'
|
425 |
if not predictions:
|
426 |
+
logger.info(f'No predictions found for {video_url}')
|
427 |
continue
|
428 |
|
429 |
# TODO use predict_args.output_as_json
|