support input ogg

src/predict.py

@@ -153,6 +153,8 @@ class Predictor:
         bucket_name = 'demovidelyuseruploads'
         if b'WAVE' in header:
             file_format = 'wav'
+        elif b'OggS' in header:
+            file_format = 'ogg'
         else:
             file_format = 'mp3'
 
@@ -163,7 +165,7 @@ class Predictor:
             file_out.write(file_bytes)
 
         wav_filename = local_filename
-        if file_format == "mp3":
+        if file_format != "wav":
             wav_filename = f"{unique_filename}.wav"
             subprocess.run(["ffmpeg", "-i", local_filename, wav_filename])
             os.remove(local_filename)

src/rp_schema.py

@@ -37,6 +37,6 @@ INPUT_VALIDATIONS = {
     'output_extension': {
         'type': str,
         'required': False,
-        'default': 'ogg'
+        'default': 'mp3'
     }
 }