Spaces:

AhmedSSabir
/

Demo-for-Gender-Score

Running

AhmedSSabir commited on Apr 19

Commit

4e74afc

•

1 Parent(s): db30631

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,8 @@ import requests
 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)