Upload wav2vec_aligen.py

wav2vec_aligen.py

@@ -28,15 +28,19 @@ def get_emissions(input_values, model):
     results = model(input_values,).logits[0]
     return results
 
+def vlaidate_range(score):
+    score = score if score <= 90 else 90
+    score = score if score >= 10 else 10
+    return score
 
 def speaker_pronunciation_assesment(audio_path):
     input_values = load_audio(audio_path, processor)
     result_scores = get_emissions(input_values, model)
 
-    pronunciation_score = round(result_scores[0].cpu().item())
-    fluency_score           = round(result_scores[1].cpu().item())
-    total_score              = round(result_scores[2].cpu().item())
-    content_scores         = round(result_scores[3].cpu().item())
+    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()))