forked from open-webui/open-webui
		
	no internet connection for whisper if you use docker
This commit is contained in:
		
							parent
							
								
									a8a0ebf3aa
								
							
						
					
					
						commit
						8ab0338e71
					
				
					 2 changed files with 11 additions and 2 deletions
				
			
		|  | @ -1,3 +1,4 @@ | |||
| import os | ||||
| from fastapi import ( | ||||
|     FastAPI, | ||||
|     Request, | ||||
|  | @ -53,12 +54,14 @@ def transcribe( | |||
|             f.write(contents) | ||||
|             f.close() | ||||
| 
 | ||||
|         model_name = WHISPER_MODEL_NAME | ||||
|         model_name = os.getenv('WHISPER_MODEL', WHISPER_MODEL_NAME) | ||||
|         download_root = os.getenv('WHISPER_DIR', f"{CACHE_DIR}/whisper/models") | ||||
| 
 | ||||
|         model = WhisperModel( | ||||
|             model_name, | ||||
|             device="cpu", | ||||
|             compute_type="int8", | ||||
|             download_root=f"{CACHE_DIR}/whisper/models", | ||||
|             download_root=download_root, | ||||
|         ) | ||||
| 
 | ||||
|         segments, info = model.transcribe(file_path, beam_size=5) | ||||
|  |  | |||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Jannik Streidl
						Jannik Streidl