From b220ccb3309b8959215e88d47e8dfd30fe1234e8 Mon Sep 17 00:00:00 2001
From: berkaybilik <bilik.berkay@gmail.com>
Date: Thu, 27 Jun 2024 00:11:06 +0100
Subject: [PATCH]  fixed reference before assignment error/warning

---
 whisper_live/server.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/whisper_live/server.py b/whisper_live/server.py
index 5815d20..749ed78 100644
--- a/whisper_live/server.py
+++ b/whisper_live/server.py
@@ -5,7 +5,7 @@ import json
 import functools
 import logging
 from enum import Enum
-from typing import List
+from typing import List, Optional
 
 import torch
 import numpy as np
@@ -156,6 +156,8 @@ class TranscriptionServer:
         self, websocket, options, faster_whisper_custom_model_path,
         whisper_tensorrt_path, trt_multilingual
     ):
+        client: Optional[ServeClientBase] = None
+
         if self.backend == "tensorrt":
             try:
                 client = ServeClientTensorRT(
@@ -196,6 +198,9 @@ class TranscriptionServer:
             )
             logging.info("Running faster_whisper backend.")
 
+        if client is None:
+            raise ValueError(f"Backend type {self.backend} not recognised or not handled.")
+
         self.client_manager.add_client(websocket, client)
 
     def get_audio_from_websocket(self, websocket):