Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
| 24 |
-
confidence_tokenizer = BertTokenizer.from_pretrained('RiteshAkhade/
|
| 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)
|