bhavyagiri commited on
Commit
73d45a9
1 Parent(s): fbfab3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,7 +9,7 @@ lang = "en"
9
 
10
  transcribe_pipe = pipeline(
11
  task="automatic-speech-recognition",
12
- model=model_ckpt,
13
  chunk_length_s=30,
14
  )
15
  transcribe_pipe.model.config.forced_decoder_ids = transcribe_pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
@@ -25,14 +25,14 @@ def yt_transcribe(yt_url):
25
  return html_embed_str, text
26
 
27
  qa_model_ckpt = "deepset/tinyroberta-squad2"
28
- qa_pipe = pipeline('question-answering', model=model_ckpt, tokenizer=model_ckpt)
29
 
30
  def get_answer(query,context):
31
  QA_input = {
32
  'question': query,
33
  'context': context
34
  }
35
- res = nlp(QA_input)["answer"]
36
  return res
37
 
38
 
 
9
 
10
  transcribe_pipe = pipeline(
11
  task="automatic-speech-recognition",
12
+ model=transcribe_model_ckpt,
13
  chunk_length_s=30,
14
  )
15
  transcribe_pipe.model.config.forced_decoder_ids = transcribe_pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
 
25
  return html_embed_str, text
26
 
27
  qa_model_ckpt = "deepset/tinyroberta-squad2"
28
+ qa_pipe = pipeline('question-answering', model=qa_model_ckpt, tokenizer=qa_model_ckpt)
29
 
30
  def get_answer(query,context):
31
  QA_input = {
32
  'question': query,
33
  'context': context
34
  }
35
+ res = qa_pipe(QA_input)["answer"]
36
  return res
37
 
38