Pendrokar commited on
Commit
c5fe670
·
1 Parent(s): b7b912d

spark top_five fix

Browse files
Files changed (1) hide show
  1. app/models.py +5 -4
app/models.py CHANGED
@@ -830,10 +830,10 @@ closed_source = [
830
  ]
831
 
832
  # top five models in order to always have one of them picked and scrutinized
833
- top_five = ['Spark-TTS']
834
 
835
  # prioritize low vote models
836
- sql = 'SELECT name FROM model WHERE (upvote + downvote) < 750 ORDER BY (upvote + downvote) ASC'
837
  conn = get_db()
838
  cursor = conn.cursor()
839
  cursor.execute(sql)
@@ -841,11 +841,12 @@ data = cursor.fetchall()
841
  for model in data:
842
  if (
843
  len(top_five) >= 5
 
 
844
  ):
845
  break
846
 
847
- if model[0] in AVAILABLE_MODELS.keys():
848
- top_five.append(model[0])
849
  print(f"low vote top_five: {top_five}")
850
 
851
  def make_link_to_space(model_name, for_leaderboard=False):
 
830
  ]
831
 
832
  # top five models in order to always have one of them picked and scrutinized
833
+ top_five = ['thunnai/SparkTTS']
834
 
835
  # prioritize low vote models
836
+ sql = 'SELECT name FROM model WHERE (upvote + downvote) < 700 ORDER BY (upvote + downvote) ASC'
837
  conn = get_db()
838
  cursor = conn.cursor()
839
  cursor.execute(sql)
 
841
  for model in data:
842
  if (
843
  len(top_five) >= 5
844
+ or model[0] in top_five
845
+ or model[0] not in AVAILABLE_MODELS.keys()
846
  ):
847
  break
848
 
849
+ top_five.append(model[0])
 
850
  print(f"low vote top_five: {top_five}")
851
 
852
  def make_link_to_space(model_name, for_leaderboard=False):