nuernie commited on
Commit
f0452ef
Β·
1 Parent(s): f37a0bf
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -2,9 +2,9 @@ import os
2
  import logging
3
  from whisper_live.server import TranscriptionServer
4
 
5
- # Set up HuggingFace cache directory in /tmp for Docker compatibility
6
  cache_dir = '/tmp/huggingface_cache'
7
- os.makedirs(cache_dir, exist_ok=True) # Ensure the directory exists
8
  os.environ['HF_HOME'] = cache_dir
9
 
10
  # β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”
@@ -14,10 +14,10 @@ logging.basicConfig(level=logging.INFO)
14
  logger = logging.getLogger(__name__)
15
 
16
  if __name__ == "__main__":
17
- # Create and start the transcription server
18
  transcription_server = TranscriptionServer()
19
  transcription_server.run(
20
  host='0.0.0.0',
21
- port=7860, # Using the default WebSocket port from the whisper_live library
22
  backend='faster_whisper'
23
  )
 
2
  import logging
3
  from whisper_live.server import TranscriptionServer
4
 
5
+ # Set up cache directory for model downloads
6
  cache_dir = '/tmp/huggingface_cache'
7
+ os.makedirs(cache_dir, exist_ok=True)
8
  os.environ['HF_HOME'] = cache_dir
9
 
10
  # β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”β€”
 
14
  logger = logging.getLogger(__name__)
15
 
16
  if __name__ == "__main__":
17
+
18
  transcription_server = TranscriptionServer()
19
  transcription_server.run(
20
  host='0.0.0.0',
21
+ port=7860,
22
  backend='faster_whisper'
23
  )