commit

app.py

@@ -12,6 +12,19 @@ SR = 22050
 N_MELS = 128
 TARGET_FRAMES = 216
 LABELS = ['cel', 'cla', 'flu', 'gac', 'gel', 'org', 'pia', 'sax', 'tru', 'vio', 'voi']
+polskie_nazwy = {
+    'cel': 'wiolonczela',
+    'cla': 'klawesyn',
+    'flu': 'flet',
+    'gac': 'gitara klasyczna',
+    'gel': 'gitara elektryczna',
+    'org': 'organy',
+    'pia': 'fortepian',
+    'sax': 'saksofon',
+    'tru': 'trąbka',
+    'vio': 'skrzypce',
+    'voi': 'głos ludzki'
+}
 
 # Wczytanie modelu
 model = tf.keras.models.load_model("model.h5")
@@ -55,8 +68,8 @@ def predict_and_plot(audio_path):
     buf.seek(0)
     image = Image.open(buf)
 
-    # Predykcje jako słownik
-    pred_dict = {label: float(p) for label, p in zip(LABELS, preds)}
+    # Predykcje jako słownik z polskimi nazwami
+    pred_dict = {polskie_nazwy[label]: float(p) for label, p in zip(LABELS, preds)}
     return pred_dict, image
 
 demo = gr.Interface(