Spaces:
Running
on
Zero
Running
on
Zero
suwonpabby
commited on
Commit
•
99a4929
1
Parent(s):
ddd4f0f
Modify
Browse files
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 |
|