Spaces:

AhmedSSabir
/

Demo-for-Gender-Score

Sleeping

AhmedSSabir commited on Apr 19

Commit

6d56bc4

•

1 Parent(s): b196c77

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,8 +18,8 @@ import numpy as np
 from sentence_transformers import SentenceTransformer, util
-#model_sts = SentenceTransformer('stsb-distilbert-base')
-model_sts = SentenceTransformer('roberta-large-nli-stsb-mean-tokens')
 #batch_size = 1
 #scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)

 from sentence_transformers import SentenceTransformer, util
+model_sts = SentenceTransformer('stsb-distilbert-base')
+#model_sts = SentenceTransformer('roberta-large-nli-stsb-mean-tokens')
 #batch_size = 1
 #scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)