feat: Add demo sharing

alexandrainst · Sep 18, 2024 · dd453ae · dd453ae
1 parent 92af25a
commit dd453ae
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/config/demo.yaml b/config/demo.yaml
@@ -7,3 +7,8 @@ title: Røst ASR Demo
 description: >
   This is a demo of the Danish speech recognition model Røst. Speak into the microphone
   and see the text appear on the screen!
+
+# Sharing parameters
+share: false
+username: null
+password: null
diff --git a/src/scripts/run_asr_demo.py b/src/scripts/run_asr_demo.py
@@ -64,7 +64,11 @@ def transcribe_audio(sampling_rate_and_audio: tuple[int, np.ndarray]) -> str:
         description=config.description,
         allow_flagging="never",
     )
-    demo.launch()
+    if config.username is not None and config.password is not None:
+        auth = (config.username, config.password)
+    else:
+        auth = None
+    demo.launch(share=config.share, auth=auth)
 
 
 if __name__ == "__main__":