Jaavid25 commited on
Commit
2adcf7c
·
verified ·
1 Parent(s): ee51323

Update modeling_apex.py

Browse files
Files changed (1) hide show
  1. modeling_apex.py +5 -5
modeling_apex.py CHANGED
@@ -205,18 +205,18 @@ class APEXModel(PreTrainedModel):
205
  for task in preds
206
  }
207
 
208
- print(f"\n{'='*50}")
209
  print(f" APEX Predictions")
210
- print(f"{'='*50}")
211
  print(f"\n Popularity:")
212
- print(f" {''*40}")
213
  print(f" {'Streams Score':<20} {results['score_streams']:>8.2f} / 100")
214
  print(f" {'Likes Score':<20} {results['score_likes']:>8.2f} / 100")
215
  print(f"\n Aesthetic Quality:")
216
- print(f" {''*40}")
217
  for dim in ["coherence", "musicality", "memorability", "clarity", "naturalness"]:
218
  print(f" {dim.capitalize():<20} {results[dim]:>8.2f} / 5.00")
219
- print(f"{'='*50}\n")
220
 
221
  if save_json:
222
  with open(save_json, "w") as f:
 
205
  for task in preds
206
  }
207
 
208
+ print(f"\n{''*50}")
209
  print(f" APEX Predictions")
210
+ print(f"{''*50}")
211
  print(f"\n Popularity:")
212
+ print(f" {'-'*40}")
213
  print(f" {'Streams Score':<20} {results['score_streams']:>8.2f} / 100")
214
  print(f" {'Likes Score':<20} {results['score_likes']:>8.2f} / 100")
215
  print(f"\n Aesthetic Quality:")
216
+ print(f" {'-'*40}")
217
  for dim in ["coherence", "musicality", "memorability", "clarity", "naturalness"]:
218
  print(f" {dim.capitalize():<20} {results[dim]:>8.2f} / 5.00")
219
+
220
 
221
  if save_json:
222
  with open(save_json, "w") as f: