RiteshAkhade commited on
Commit
c406962
·
verified ·
1 Parent(s): 79c919b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/confidence_model')
15
- confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/confidence_model')
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')