cfc-tech commited on
Commit
b8c5f6f
·
verified ·
1 Parent(s): 7912bc7
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -2,10 +2,14 @@ import streamlit as st
2
  from pytube import YouTube
3
  from moviepy.editor import *
4
  import torch
5
- from transformers import Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer
6
  from transformers import pipeline
7
  import librosa
8
 
 
 
 
 
9
  # Streamlit interface setup
10
  st.title("YouTube Video Summarizer")
11
 
 
2
  from pytube import YouTube
3
  from moviepy.editor import *
4
  import torch
5
+ from transformers import Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer, pipeline, logging
6
  from transformers import pipeline
7
  import librosa
8
 
9
+
10
+ # Suppress warnings from transformers
11
+ logging.set_verbosity_error()
12
+
13
  # Streamlit interface setup
14
  st.title("YouTube Video Summarizer")
15