Update main.py

main.py

@@ -109,7 +109,16 @@ async def upload_audio(file: UploadFile = File(...)):
 
     # Tokenization
     print("Tokenizing audio...")
-    input_values = await asyncio.to_thread(tokenizer, audio, return_tensors="pt")
+    input_values = tokenizer(
+        audio, 
+        return_tensors="pt", 
+        padding="max_length", 
+        max_length= 1000000, 
+        truncation=True
+    ).input_values
+
+    print(input_values.shape)
+    
     input_values = input_values.input_values
     print("Tokenization complete. Shape of input_values:", input_values.shape)