suwonpabby commited on
Commit
99a4929
1 Parent(s): ddd4f0f
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -185,7 +185,9 @@ def make_gen(QA, candidates, top_k, character_type):
185
  batch_size=1,
186
  max_length=8192,
187
  )
188
-
 
 
189
  key_embeddings = embedding_model.encode(candidates)["dense_vecs"]
190
 
191
  similarity = query_embeddings @ key_embeddings.T
@@ -245,8 +247,6 @@ async def root_endpoint(request: Request):
245
  candidates = data.get("candidates")
246
  top_k = data.get("top_k")
247
  character_type = data.get("character_type")
248
- print(QA)
249
- print(candidates)
250
  return StreamingResponse(gen_stream(QA, candidates, top_k, character_type), media_type="text/event-stream")
251
 
252
 
 
185
  batch_size=1,
186
  max_length=8192,
187
  )
188
+ print("⭐️")
189
+ print(type(candidates))
190
+ print(type(candidates[0]))
191
  key_embeddings = embedding_model.encode(candidates)["dense_vecs"]
192
 
193
  similarity = query_embeddings @ key_embeddings.T
 
247
  candidates = data.get("candidates")
248
  top_k = data.get("top_k")
249
  character_type = data.get("character_type")
 
 
250
  return StreamingResponse(gen_stream(QA, candidates, top_k, character_type), media_type="text/event-stream")
251
 
252