Update main.py

main.py

@@ -3,6 +3,7 @@ import datetime
 import time
 import torch
 import os
+import numpy as np
 from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC, Wav2Vec2ProcessorWithLM, AutoConfig
 from huggingface_hub import hf_hub_download
 from fuzzywuzzy import fuzz
@@ -101,6 +102,9 @@ async def predict(
 
     """
     upload_audio = ffmpeg_read(file, sampling_rate=16000)
+    print(upload_audio.shape)
+    print(np.sum(np.abs(upload_audio)))
+    debug = np.sum(np.abs(upload_audio))               
     audio_duration = len(upload_audio) / 16000
     current_time = datetime.datetime.now().strftime("%Y-%h-%d-%H:%M:%S")
     start_time = time.time()
@@ -144,5 +148,6 @@ async def predict(
             "pitch predict": best_pitch_predict,
             "wrong word index": wrong_word,
             "wrong pitch index": wrong_pitch,
-            "score": score
+            "score": score,
+            "debug": debug,
             }