fixed typo and added amr support

GooeyAI · Oct 23, 2023 · 06ed3ea · 06ed3ea
1 parent 2a40f15
commit 06ed3ea
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/daras_ai_v2/asr.py b/daras_ai_v2/asr.py
@@ -294,7 +294,7 @@ def run_asr(
     is_youtube_url = "youtube" in audio_url or "youtu.be" in audio_url
     if is_youtube_url:
         audio_url, size = download_youtube_to_wav(audio_url)
-    elif is_gdrive_url:
+    elif is_gdrive_url(furl(audio_url)):
         meta: dict[str, str] = gdrive_metadata(url_to_gdrive_file_id(furl(audio_url)))
         anybytes, ext = gdrive_download(
             furl(audio_url), meta.get("mimeType", "audio/wav")

diff --git a/recipes/asr.py b/recipes/asr.py
@@ -80,7 +80,17 @@ def related_workflows(self) -> list:
     def render_form_v2(self):
         document_uploader(
             "##### Audio Files",
-            accept=(".wav", ".ogg", ".mp3", ".aac", ".opus", ".oga", ".mp4", ".webm"),
+            accept=(
+                ".wav",
+                ".ogg",
+                ".mp3",
+                ".aac",
+                ".opus",
+                ".oga",
+                ".mp4",
+                ".webm",
+                ".amr",
+            ),
         )
         col1, col2 = st.columns(2, responsive=False)
         with col1: