don0726 commited on
Commit
7895d09
·
verified ·
1 Parent(s): fb2860e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,14 +8,14 @@ from typing import List
8
  app = FastAPI()
9
 
10
  # ===== CONFIGURATION =====
11
- MAX_BATCH_SIZE = 8 # 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
- "small",
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