Shahadbal commited on
Commit
77e4bf5
·
verified ·
1 Parent(s): 6f180fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=model, tokenizer=tokenizer, qg_format="prepend")
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"):