jsd219 commited on
Commit
3195410
·
verified ·
1 Parent(s): c8a8e04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -19,18 +19,16 @@ def estimate_genre(filepath):
19
  return {"Error": f"Feature extraction failed: {str(e)}"}
20
 
21
  # Heuristic genre rules
22
- if tempo > 140 and centroid > 3000:
23
  genre = "Electronic / Dance"
24
- elif tempo < 80 and chroma_strength < 0.5:
25
- genre = "Classical"
26
- elif 90 < tempo < 110:
27
- genre = "Hip Hop / R&B"
28
- elif 110 <= tempo <= 130:
29
  genre = "Pop / Rock"
30
- elif tempo > 130 and chroma_strength > 0.6:
31
- genre = "Funk / Disco"
32
  else:
33
- genre = "Other"
34
 
35
  return {
36
  "Predicted Genre": genre,
 
19
  return {"Error": f"Feature extraction failed: {str(e)}"}
20
 
21
  # Heuristic genre rules
22
+ if tempo > 140 and brightness > 2500:
23
  genre = "Electronic / Dance"
24
+ elif tempo < 90 and chroma_strength < 0.3:
25
+ genre = "Hip Hop / Trap"
26
+ elif 100 < tempo < 130 and brightness > 2300 and chroma_strength > 0.7:
 
 
27
  genre = "Pop / Rock"
28
+ elif brightness < 1800 and chroma_strength > 0.9:
29
+ genre = "Jazz / Soul"
30
  else:
31
+ genre = "Unknown"
32
 
33
  return {
34
  "Predicted Genre": genre,