Update main.py

main.py

@@ -107,7 +107,6 @@ async def upload_audio(file: UploadFile = File(...)):
 
     print("audio:" , audio[:5])
     
-    return "audio sample extracted:" 
     
 
     print("length of the audio array:" , len(audio))
@@ -120,7 +119,7 @@ async def upload_audio(file: UploadFile = File(...)):
         audio, 
         return_tensors="pt", 
         padding="max_length", 
-        max_length= 1000000, 
+        max_length= 386380, 
         truncation=True
     ).input_values
 
@@ -128,6 +127,8 @@ async def upload_audio(file: UploadFile = File(...)):
     
     print("Tokenization complete. Shape of input_values:", input_values.shape)
 
+    return "tokenization successful" 
+    
     # Perform inference
     print("Performing inference with Wav2Vec2 model...")