use tinyroberta for qa
Browse files
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 |
-
|
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
|