Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- app/main.py +4 -4
- models/hear_classifier_v9_ultimate.h5 +3 -0
- test_real_pipeline.py +2 -2
app/main.py
CHANGED
|
@@ -52,7 +52,7 @@ app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 16MB limit
|
|
| 52 |
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
|
| 53 |
|
| 54 |
# Configuration
|
| 55 |
-
MODEL_PATH = os.path.join(project_root, "models", "
|
| 56 |
|
| 57 |
# Global variables for lazy loading
|
| 58 |
extractor = None
|
|
@@ -66,15 +66,15 @@ def load_resources():
|
|
| 66 |
extractor = HeARExtractor(token=hf_token)
|
| 67 |
|
| 68 |
if classifier_model is None:
|
| 69 |
-
print(f"Loading
|
| 70 |
classifier_model = load_model(MODEL_PATH, compile=False)
|
| 71 |
|
| 72 |
@app.route('/')
|
| 73 |
def index():
|
| 74 |
return jsonify({
|
| 75 |
"status": "online",
|
| 76 |
-
"service": "KasaHealth Diagnostic API (
|
| 77 |
-
"version": "1.
|
| 78 |
"message": "Send audio files via POST to /predict"
|
| 79 |
})
|
| 80 |
|
|
|
|
| 52 |
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
|
| 53 |
|
| 54 |
# Configuration
|
| 55 |
+
MODEL_PATH = os.path.join(project_root, "models", "hear_classifier_v9_ultimate.h5")
|
| 56 |
|
| 57 |
# Global variables for lazy loading
|
| 58 |
extractor = None
|
|
|
|
| 66 |
extractor = HeARExtractor(token=hf_token)
|
| 67 |
|
| 68 |
if classifier_model is None:
|
| 69 |
+
print(f"Loading Ultimate Model from {MODEL_PATH}...")
|
| 70 |
classifier_model = load_model(MODEL_PATH, compile=False)
|
| 71 |
|
| 72 |
@app.route('/')
|
| 73 |
def index():
|
| 74 |
return jsonify({
|
| 75 |
"status": "online",
|
| 76 |
+
"service": "KasaHealth Diagnostic API (Ultimate V9)",
|
| 77 |
+
"version": "1.3.0",
|
| 78 |
"message": "Send audio files via POST to /predict"
|
| 79 |
})
|
| 80 |
|
models/hear_classifier_v9_ultimate.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:625a0f1facd74277cdcb33420a5feacf79e8caf2a2b3ce68d052ad199d5bf62a
|
| 3 |
+
size 14685688
|
test_real_pipeline.py
CHANGED
|
@@ -19,7 +19,7 @@ from utils.audio_preprocessor import advanced_preprocess
|
|
| 19 |
AUDIO_ROOT = r"c:\Users\ASUS\lung_ai_project\data\coughvid_public\organized"
|
| 20 |
SICK_DIR = os.path.join(AUDIO_ROOT, "sick")
|
| 21 |
HEALTHY_DIR = os.path.join(AUDIO_ROOT, "healthy")
|
| 22 |
-
MODEL_PATH = os.path.join(PROJECT_ROOT, "models", "
|
| 23 |
|
| 24 |
META_PATH = r"c:\Users\ASUS\lung_ai_project\data\coughvid_public\metadata_compiled.csv"
|
| 25 |
import pandas as pd
|
|
@@ -89,7 +89,7 @@ def test_pipeline():
|
|
| 89 |
prob = classifier_model.predict(X, verbose=0)[0][0]
|
| 90 |
|
| 91 |
# 6. API Logic
|
| 92 |
-
THRESHOLD = 0.
|
| 93 |
if prob > THRESHOLD:
|
| 94 |
final_label = "sick"
|
| 95 |
else:
|
|
|
|
| 19 |
AUDIO_ROOT = r"c:\Users\ASUS\lung_ai_project\data\coughvid_public\organized"
|
| 20 |
SICK_DIR = os.path.join(AUDIO_ROOT, "sick")
|
| 21 |
HEALTHY_DIR = os.path.join(AUDIO_ROOT, "healthy")
|
| 22 |
+
MODEL_PATH = os.path.join(PROJECT_ROOT, "models", "hear_classifier_v9_ultimate.h5")
|
| 23 |
|
| 24 |
META_PATH = r"c:\Users\ASUS\lung_ai_project\data\coughvid_public\metadata_compiled.csv"
|
| 25 |
import pandas as pd
|
|
|
|
| 89 |
prob = classifier_model.predict(X, verbose=0)[0][0]
|
| 90 |
|
| 91 |
# 6. API Logic
|
| 92 |
+
THRESHOLD = 0.50
|
| 93 |
if prob > THRESHOLD:
|
| 94 |
final_label = "sick"
|
| 95 |
else:
|