jayesh95 commited on
Commit
ed6b579
1 Parent(s): ee54e23

use tinyroberta for qa

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ model = whisper.load_model("base", device = device)
24
  model_med = whisper.load_model("medium", device = device)
25
 
26
  #Roberta Q&A
27
- #model_name = "deepset/tinyroberta-squad2"
28
- model_name = "deepset/roberta-base-squad2"
29
  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name, device = 0 if device == "cuda" else -1)
30
 
31
  #TTS
 
24
  model_med = whisper.load_model("medium", device = device)
25
 
26
  #Roberta Q&A
27
+ model_name = "deepset/tinyroberta-squad2"
28
+ #model_name = "deepset/roberta-base-squad2"
29
  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name, device = 0 if device == "cuda" else -1)
30
 
31
  #TTS