Spaces:

legacy107
/

flan-t5-large-ia3-wiki

Runtime error

legacy107 commited on Oct 23, 2023

Commit

7717dbf

1 Parent(s): 751e496

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ def retrieve_context(query, contexts):
     corpus_embeddings = bi_encoder.encode(contexts, convert_to_tensor=True, show_progress_bar=False)
     question_embedding = bi_encoder.encode(query, convert_to_tensor=True, show_progress_bar=False)
-    question_embedding = question_embedding.cuda()
     hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=top_k)
     hits = hits[0]

     corpus_embeddings = bi_encoder.encode(contexts, convert_to_tensor=True, show_progress_bar=False)
     question_embedding = bi_encoder.encode(query, convert_to_tensor=True, show_progress_bar=False)
+    question_embedding = question_embedding
     hits = util.semantic_search(question_embedding, corpus_embeddings, top_k=top_k)
     hits = hits[0]