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 |
)
|