Spaces:

AhmedSSabir
/

Demo-for-Gender-Score

Running

AhmedSSabir commited on Oct 29, 2023

Commit

2b94d4c

•

1 Parent(s): a571a7b

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)