Spaces:

sergiomar73
/

qc-nlp-003-similarity-with-roberta

Runtime error

sergiomar73 commited on Sep 29, 2022

Commit

4bd1056

•

1 Parent(s): cdffa75

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 from sentence_transformers import SentenceTransformer, util
-model = SentenceTransformer('all-roberta-large-v1')
 def calculate_embeddings_with_roberta(text):
   embeddings = model.encode(text, convert_to_tensor=True)

 import os
 from sentence_transformers import SentenceTransformer, util
+model = SentenceTransformer('paraphrase-distilroberta-base-v2')
 def calculate_embeddings_with_roberta(text):
   embeddings = model.encode(text, convert_to_tensor=True)