submission-frugal-ai-challenge

Running

pedro-thenewsroom commited on Jan 31

Commit

f193c7f

verified ·

1 Parent(s): 6d303d4

Update tasks/text.py

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -14,8 +14,8 @@ DESCRIPTION = "Efficient embedding-based classification with similarity threshol
 ROUTE = "/text"
 # Load custom embedding model
-# model_name = "pedro-thenewsroom/climate-misinfo-embed"
-model_name = "pedro-thenewsroom/climate-misinfo-embed-8bit"
 embedding_model = SentenceTransformer(model_name)
 # Define label mapping

 ROUTE = "/text"
 # Load custom embedding model
+model_name = "pedro-thenewsroom/climate-misinfo-embed"
+# model_name = "pedro-thenewsroom/climate-misinfo-embed-8bit"
 embedding_model = SentenceTransformer(model_name)
 # Define label mapping