ajanz commited on
Commit
165779a
·
1 Parent(s): 88d581a

testing padding impact (just for debug - will be removed later)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ index = load_index()
43
 
44
 
45
  def predict(text: str = sample_text, top_k: int=3):
46
- text = text + "".join(['[PAD]' * 252])
47
  index_data, faiss_index = index
48
  # takes only the [CLS] embedding (for now)
49
  query = model(text, return_tensors='pt')[0][0].numpy().reshape(1, -1)
 
43
 
44
 
45
  def predict(text: str = sample_text, top_k: int=3):
46
+ text = text + "".join(['[PAD]'] * 252)
47
  index_data, faiss_index = index
48
  # takes only the [CLS] embedding (for now)
49
  query = model(text, return_tensors='pt')[0][0].numpy().reshape(1, -1)