stogaja commited on
Commit
83db0fb
·
1 Parent(s): 5c40b85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,14 +65,14 @@ with mod_container:
65
  # let's pass the input to the loaded_model with torch compiled with cuda
66
  if prompt:
67
  # let's get the result
68
- simscore = PathFinder.predict([prompt])
69
 
70
  from sentence_transformers import CrossEncoder
71
  loaded_model = CrossEncoder("cross-encoder/stsb-roberta-base")
72
  sentence_pairs = []
73
  for sentence1, sentence2 in zip(data['sentence1'], data['sentence2']):
74
  sentence_pairs.append([sentence1, sentence2])
75
-
 
76
  # sorting the df to get highest scoring xpath_container
77
  data['SBERT CrossEncoder_Score'] = loaded_model.predict(sentence_pairs)
78
  most_acc = data.head(5)
 
65
  # let's pass the input to the loaded_model with torch compiled with cuda
66
  if prompt:
67
  # let's get the result
 
68
 
69
  from sentence_transformers import CrossEncoder
70
  loaded_model = CrossEncoder("cross-encoder/stsb-roberta-base")
71
  sentence_pairs = []
72
  for sentence1, sentence2 in zip(data['sentence1'], data['sentence2']):
73
  sentence_pairs.append([sentence1, sentence2])
74
+
75
+ simscore = PathFinder.predict([prompt])
76
  # sorting the df to get highest scoring xpath_container
77
  data['SBERT CrossEncoder_Score'] = loaded_model.predict(sentence_pairs)
78
  most_acc = data.head(5)