RiteshAkhade commited on
Commit
e011761
·
verified ·
1 Parent(s): 2d7d058

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ from app.questions import get_question
20
 
21
  # Load models
22
  whisper_model = whisper.load_model("small")
23
- confidence_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/confidence_model1')
24
- confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/confidence_tokenizer1')
25
  context_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/context_model')
26
  context_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/context_model')
27
  emotion_pipe = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
 
20
 
21
  # Load models
22
  whisper_model = whisper.load_model("small")
23
+ confidence_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/final_confidence')
24
+ confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/final_confidence')
25
  context_model = BertForSequenceClassification.from_pretrained('RiteshAkhade/context_model')
26
  context_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/context_model')
27
  emotion_pipe = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)