Upload models.py with huggingface_hub

models.py

@@ -79,6 +79,7 @@ class Vote(db.Model):
     # Sentence origin tracking
     sentence_hash = db.Column(db.String(64), nullable=True, index=True)  # SHA-256 hash of the sentence
     sentence_origin = db.Column(db.String(20), nullable=True)  # 'dataset', 'custom', 'unknown'
+    sentence_category = db.Column(db.String(50), nullable=True)  # 'service', 'podcast', 'diacritics', 'user-generated', etc.
     counts_for_public_leaderboard = db.Column(db.Boolean, default=True)  # Whether this vote counts for public leaderboard
 
     chosen = db.relationship(
@@ -233,9 +234,10 @@ def anonymize_ip_address(ip_address):
         return None
 
 
-def record_vote(user_id, text, chosen_model_id, rejected_model_id, model_type, 
-                session_duration=None, ip_address=None, user_agent=None, 
-                generation_date=None, cache_hit=None, all_dataset_sentences=None):
+def record_vote(user_id, text, chosen_model_id, rejected_model_id, model_type,
+                session_duration=None, ip_address=None, user_agent=None,
+                generation_date=None, cache_hit=None, all_dataset_sentences=None,
+                sentence_category=None):
     """Record a vote and update Elo ratings."""
     
     # Determine sentence origin and whether it should count for public leaderboard
@@ -266,6 +268,7 @@ def record_vote(user_id, text, chosen_model_id, rejected_model_id, model_type,
         cache_hit=cache_hit,
         sentence_hash=sentence_hash,
         sentence_origin=sentence_origin,
+        sentence_category=sentence_category if sentence_category else ('user-generated' if sentence_origin == 'custom' else None),
         counts_for_public_leaderboard=counts_for_public,
     )
     db.session.add(vote)
@@ -634,7 +637,7 @@ def insert_initial_models():
             is_active=True,
             model_url="https://inworld.ai/tts",
             voice="Ashley",
-            api_model_id="inworld-tts-1.5",
+            api_model_id="inworld-tts-1",
         ),
         Model(
             id="inworld-max",