Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,8 @@ from app.questions import get_question
|
|
11 |
whisper_model = whisper.load_model("small")
|
12 |
|
13 |
# Load the pre-trained model and tokenizer for confidence analysis
|
14 |
-
confidence_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/
|
15 |
-
confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/
|
16 |
|
17 |
# Load pre-trained context analysis model (BERT-based)
|
18 |
context_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/context_model')
|
|
|
11 |
whisper_model = whisper.load_model("small")
|
12 |
|
13 |
# Load the pre-trained model and tokenizer for confidence analysis
|
14 |
+
confidence_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/Confidence_model2')
|
15 |
+
confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/Confidence_model2')
|
16 |
|
17 |
# Load pre-trained context analysis model (BERT-based)
|
18 |
context_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/context_model')
|