Yilin0601 commited on
Commit
20b8be9
·
verified ·
1 Parent(s): 3e63959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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-accuracy-checkpoints"
14
  )
15
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(
16
- "Yilin0601/wav2vec2-accuracy-checkpoints"
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 Accuracy Predictor",
68
  description=(
69
- "This demo uses a fine-tuned Wav2Vec2ForSequenceClassification model with labels "
70
- "mapped from 3..10. Record or upload audio to see the predicted level."
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
  )