AhmedSSabir commited on
Commit
2d3cd81
1 Parent(s): 5ac0349

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -13,12 +13,10 @@ from sentence_transformers import SentenceTransformer, util
13
  from sklearn.metrics.pairwise import cosine_similarity
14
 
15
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
16
- model = gr.interface.huggingface.load('stsb-distilbert-base')
17
 
18
  #SentenceTransformer('stsb-distilbert-base', device=device)
19
 
20
-
21
-
22
  batch_size = 1
23
  scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)
24
 
 
13
  from sklearn.metrics.pairwise import cosine_similarity
14
 
15
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
16
+ model = gr.interface.huggingface.load('sentence-transformers/stsb-distilbert-base')
17
 
18
  #SentenceTransformer('stsb-distilbert-base', device=device)
19
 
 
 
20
  batch_size = 1
21
  scorer = LMScorer.from_pretrained('gpt2' , device=device, batch_size=batch_size)
22