AliJabbarpour commited on
Commit
7cb2c54
·
verified ·
1 Parent(s): 07bf71d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -8,6 +8,16 @@ import threading
8
  import torch
9
  import torchaudio
10
  from fastapi import FastAPI, Request, HTTPException, BackgroundTasks
 
 
 
 
 
 
 
 
 
 
11
  from fastapi.middleware.cors import CORSMiddleware
12
  from faster_whisper import WhisperModel
13
  from df.enhance import enhance, init_df, load_audio, save_audio
 
8
  import torch
9
  import torchaudio
10
  from fastapi import FastAPI, Request, HTTPException, BackgroundTasks
11
+
12
+ class HealthAndStatusFilter(logging.Filter):
13
+ """جلوی چاپ لاگ‌های /status/ و /health رو می‌گیره تا کنسول شلوغ نشه (خودِ درخواست‌ها عادی پردازش میشن)"""
14
+ def filter(self, record):
15
+ message = record.getMessage()
16
+ return "/status/" not in message and "/health" not in message
17
+
18
+
19
+ logging.getLogger("uvicorn.access").addFilter(HealthAndStatusFilter())
20
+
21
  from fastapi.middleware.cors import CORSMiddleware
22
  from faster_whisper import WhisperModel
23
  from df.enhance import enhance, init_df, load_audio, save_audio