Update wav2vec_aligen.py

wav2vec_aligen.py

@@ -28,13 +28,13 @@ def get_emissions(input_values, model):
     results = model(input_values,).logits[0]
     return results
 
-def validate_range(value):
+def validate_range(value, minus):
     value = 10 + value * 80
     if value > 90:
         return 90
         
     if value < 50 and value > 10:
-        value = value - 5
+        value = value - minus
     elif value < 10:
         value = 10
     else: None
@@ -44,10 +44,10 @@ def speaker_pronunciation_assesment(audio_path):
     input_values = load_audio(audio_path, processor)
     result_scores = get_emissions(input_values, model)
 
-    pronunciation_score = vlaidate_range(round(result_scores[0].cpu().item()))
-    fluency_score           = vlaidate_range(round(result_scores[1].cpu().item()))
-    total_score               = vlaidate_range(round(result_scores[2].cpu().item()))
-    content_scores         = vlaidate_range(round(result_scores[3].cpu().item()))
+    pronunciation_score = vlaidate_range(round(result_scores[0].cpu().item()), 5)
+    fluency_score       = vlaidate_range(round(result_scores[1].cpu().item()), 12)
+    total_score         = vlaidate_range(round(result_scores[2].cpu().item()), 5)
+    content_scores      = vlaidate_range(round(result_scores[3].cpu().item()), 7)