Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -49,10 +49,10 @@ from pipelines import pipeline
|
|
49 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
50 |
|
51 |
# Load the tokenizer and model
|
52 |
-
tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-small-qg-prepend", use_fast=False) # Disable the fast tokenizer
|
53 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-small-qg-prepend")
|
54 |
|
55 |
-
question_generator = pipeline("question-generation", model=
|
56 |
|
57 |
# Function to download audio from YouTube
|
58 |
def download_audio_from_youtube(youtube_url, output_path="downloaded_audio.mp3"):
|
|
|
49 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
50 |
|
51 |
# Load the tokenizer and model
|
52 |
+
# tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-small-qg-prepend", use_fast=False) # Disable the fast tokenizer
|
53 |
+
# model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-small-qg-prepend")
|
54 |
|
55 |
+
question_generator = pipeline("question-generation", model="valhalla/t5-small-qg-prepend" qg_format="prepend")
|
56 |
|
57 |
# Function to download audio from YouTube
|
58 |
def download_audio_from_youtube(youtube_url, output_path="downloaded_audio.mp3"):
|