Update app.py
Browse files
app.py
CHANGED
|
@@ -8,14 +8,14 @@ from typing import List
|
|
| 8 |
app = FastAPI()
|
| 9 |
|
| 10 |
# ===== CONFIGURATION =====
|
| 11 |
-
MAX_BATCH_SIZE =
|
| 12 |
QUEUE_TIMEOUT = 0.1 # wait time to collect batch
|
| 13 |
MODEL_WORKERS = 2 # your 2 vCPU
|
| 14 |
|
| 15 |
# ===== LOAD MODELS =====
|
| 16 |
models = [
|
| 17 |
WhisperModel(
|
| 18 |
-
"
|
| 19 |
device="cpu",
|
| 20 |
compute_type="int8",
|
| 21 |
cpu_threads=1
|
|
|
|
| 8 |
app = FastAPI()
|
| 9 |
|
| 10 |
# ===== CONFIGURATION =====
|
| 11 |
+
MAX_BATCH_SIZE = 4 # max requests processed together
|
| 12 |
QUEUE_TIMEOUT = 0.1 # wait time to collect batch
|
| 13 |
MODEL_WORKERS = 2 # your 2 vCPU
|
| 14 |
|
| 15 |
# ===== LOAD MODELS =====
|
| 16 |
models = [
|
| 17 |
WhisperModel(
|
| 18 |
+
"base",
|
| 19 |
device="cpu",
|
| 20 |
compute_type="int8",
|
| 21 |
cpu_threads=1
|