PBusienei commited on
Commit
8e36b55
1 Parent(s): 6143265

remove astype float

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def main():
61
  #st.text_area('Text area')
62
  #age = st.number_input("Age in Years")
63
  #Encode query and documents
64
- query_emb = model(query).astype(float)
65
 
66
  #Compute dot score between query and all document embeddings
67
  #scores = util.dot_score(query_emb, doc_emb.astype(float))#[0].cpu().tolist()
 
61
  #st.text_area('Text area')
62
  #age = st.number_input("Age in Years")
63
  #Encode query and documents
64
+ query_emb = model(query)#.astype(float)
65
 
66
  #Compute dot score between query and all document embeddings
67
  #scores = util.dot_score(query_emb, doc_emb.astype(float))#[0].cpu().tolist()