Update app.py
Browse files
app.py
CHANGED
|
@@ -10,10 +10,10 @@ from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2FeatureExtra
|
|
| 10 |
# This model was fine-tuned with labels remapped from [3..10] to [0..7].
|
| 11 |
# Make sure the model repo name below is correct and accessible.
|
| 12 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(
|
| 13 |
-
"Yilin0601/wav2vec2-
|
| 14 |
)
|
| 15 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(
|
| 16 |
-
"Yilin0601/wav2vec2-
|
| 17 |
)
|
| 18 |
|
| 19 |
# --------------------------------------------------
|
|
@@ -64,10 +64,10 @@ iface = gr.Interface(
|
|
| 64 |
fn=predict,
|
| 65 |
inputs=gr.Audio(type="numpy", label="Record or Upload Audio"),
|
| 66 |
outputs="text",
|
| 67 |
-
title="L2 English
|
| 68 |
description=(
|
| 69 |
-
"This demo uses a fine-tuned Wav2Vec2ForSequenceClassification model with labels "
|
| 70 |
-
"mapped from
|
| 71 |
),
|
| 72 |
allow_flagging="never"
|
| 73 |
)
|
|
|
|
| 10 |
# This model was fine-tuned with labels remapped from [3..10] to [0..7].
|
| 11 |
# Make sure the model repo name below is correct and accessible.
|
| 12 |
model = Wav2Vec2ForSequenceClassification.from_pretrained(
|
| 13 |
+
"Yilin0601/wav2vec2-fluency-checkpoints"
|
| 14 |
)
|
| 15 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(
|
| 16 |
+
"Yilin0601/wav2vec2-fluency-checkpoints"
|
| 17 |
)
|
| 18 |
|
| 19 |
# --------------------------------------------------
|
|
|
|
| 64 |
fn=predict,
|
| 65 |
inputs=gr.Audio(type="numpy", label="Record or Upload Audio"),
|
| 66 |
outputs="text",
|
| 67 |
+
title="L2 English Fluency Predictor",
|
| 68 |
description=(
|
| 69 |
+
"This demo uses a fine-tuned Wav2Vec2ForSequenceClassification model with labels for accuracy evaluation "
|
| 70 |
+
"mapped from 0 to 10. Record or upload audio to see the predicted level."
|
| 71 |
),
|
| 72 |
allow_flagging="never"
|
| 73 |
)
|