Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,6 +14,10 @@ WHISPER_MODEL_SIZE = "base"
|
|
| 14 |
# of performance and size for summarization.
|
| 15 |
SUMMARIZATION_MODEL = "sshleifer/distilbart-cnn-12-6"
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
# --- Load Models (once at startup) ---
|
| 18 |
# Load the Whisper ASR model
|
| 19 |
print(f"Loading Whisper model: {WHISPER_MODEL_SIZE}...")
|
|
@@ -49,6 +53,14 @@ def download_and_extract_audio(youtube_url):
|
|
| 49 |
'no_warnings': True,
|
| 50 |
}
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
try:
|
| 53 |
print(f"Downloading audio for {youtube_url} to {audio_path}...")
|
| 54 |
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
@@ -126,8 +138,11 @@ iface = gr.Interface(
|
|
| 126 |
description=(
|
| 127 |
"Enter a YouTube video URL, and this tool will download its audio, "
|
| 128 |
"transcribe it using OpenAI Whisper, and then generate a summary/notes."
|
| 129 |
-
"
|
| 130 |
-
"
|
|
|
|
|
|
|
|
|
|
| 131 |
),
|
| 132 |
allow_flagging="auto", # Allows users to flag results for review
|
| 133 |
examples=[
|
|
|
|
| 14 |
# of performance and size for summarization.
|
| 15 |
SUMMARIZATION_MODEL = "sshleifer/distilbart-cnn-12-6"
|
| 16 |
|
| 17 |
+
# Define the path to the cookies file. Make sure you have uploaded 'cookies.txt'
|
| 18 |
+
# to the root directory of your Hugging Face Space.
|
| 19 |
+
COOKIES_FILE_PATH = "cookies.txt"
|
| 20 |
+
|
| 21 |
# --- Load Models (once at startup) ---
|
| 22 |
# Load the Whisper ASR model
|
| 23 |
print(f"Loading Whisper model: {WHISPER_MODEL_SIZE}...")
|
|
|
|
| 53 |
'no_warnings': True,
|
| 54 |
}
|
| 55 |
|
| 56 |
+
# Add cookies if the file exists
|
| 57 |
+
if os.path.exists(COOKIES_FILE_PATH):
|
| 58 |
+
ydl_opts['cookiefile'] = COOKIES_FILE_PATH
|
| 59 |
+
print(f"Using cookies from {COOKIES_FILE_PATH}")
|
| 60 |
+
else:
|
| 61 |
+
print(f"Cookies file not found at {COOKIES_FILE_PATH}. Proceeding without cookies.")
|
| 62 |
+
|
| 63 |
+
|
| 64 |
try:
|
| 65 |
print(f"Downloading audio for {youtube_url} to {audio_path}...")
|
| 66 |
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
|
|
| 138 |
description=(
|
| 139 |
"Enter a YouTube video URL, and this tool will download its audio, "
|
| 140 |
"transcribe it using OpenAI Whisper, and then generate a summary/notes."
|
| 141 |
+
"<br><b>Note:</b> If you encounter download issues (e.g., 'Sign in to confirm you’re not a bot'), "
|
| 142 |
+
"ensure you have uploaded a `cookies.txt` file (exported from your browser) to the root of this Hugging Face Space."
|
| 143 |
+
"<br><b>Performance:</b> Analyzing long videos (e.g., 1 hour+) can take a significant amount of time "
|
| 144 |
+
"and consume considerable resources, especially on CPU-only Hugging Face Spaces. "
|
| 145 |
+
"For faster processing of long videos, consider upgrading your Hugging Face Space to a GPU instance."
|
| 146 |
),
|
| 147 |
allow_flagging="auto", # Allows users to flag results for review
|
| 148 |
examples=[
|