Commit @a76f247f9dfb40d41a942b418f6b428705cbc393 - yjyoon/whisper_server

Y
List All
Feedback
- This Project
- All Projects

Fedir Zadniprovskyi 2024-07-16

fix: lint errors

@a76f247f9dfb40d41a942b418f6b428705cbc393

5b12f91

a76f247

faster_whisper_server/gradio_app.py

--- faster_whisper_server/gradio_app.py

+++ faster_whisper_server/gradio_app.py


 
 TRANSCRIPTION_ENDPOINT = "/v1/audio/transcriptions"
 TRANSLATION_ENDPOINT = "/v1/audio/translations"
+TIMEOUT_SECONDS = 180
 
 
 def create_gradio_demo(config: Config) -> gr.Blocks:
     host = os.getenv("UVICORN_HOST", "0.0.0.0")
     port = int(os.getenv("UVICORN_PORT", "8000"))
     # NOTE: worth looking into generated clients
-    http_client = httpx.Client(base_url=f"http://{host}:{port}", timeout=None)
+    http_client = httpx.Client(base_url=f"http://{host}:{port}", timeout=httpx.Timeout(timeout=TIMEOUT_SECONDS))
     openai_client = OpenAI(base_url=f"http://{host}:{port}/v1", api_key="cant-be-empty")
 
     def handler(file_path: str, model: str, task: Task, temperature: float, stream: bool) -> Generator[str, None, None]:

Add a comment

Open 0
Closed 0

List

...	...	@@ -10,13 +10,14 @@
10	10
11	11	TRANSCRIPTION_ENDPOINT = "/v1/audio/transcriptions"
12	12	TRANSLATION_ENDPOINT = "/v1/audio/translations"
	13	+TIMEOUT_SECONDS = 180
13	14
14	15
15	16	def create_gradio_demo(config: Config) -> gr.Blocks:
16	17	host = os.getenv("UVICORN_HOST", "0.0.0.0")
17	18	port = int(os.getenv("UVICORN_PORT", "8000"))
18	19	# NOTE: worth looking into generated clients
19		- http_client = httpx.Client(base_url=f"http://{host}:{port}", timeout=None)
	20	+ http_client = httpx.Client(base_url=f"http://{host}:{port}", timeout=httpx.Timeout(timeout=TIMEOUT_SECONDS))
20	21	openai_client = OpenAI(base_url=f"http://{host}:{port}/v1", api_key="cant-be-empty")
21	22
22	23	def handler(file_path: str, model: str, task: Task, temperature: float, stream: bool) -> Generator[str, None, None]:

Delete comment