Spaces:
Sleeping
Sleeping
Commit
·
c15ec08
1
Parent(s):
b51f89f
Update app.py
Browse files
app.py
CHANGED
@@ -593,20 +593,7 @@ async def resources_endpoint(profile: MedicalProfile):
|
|
593 |
scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
|
594 |
scored_documents = list(zip(scores, document_ids, document_texts))
|
595 |
scored_documents.sort(key=lambda x: x[0], reverse=True)
|
596 |
-
|
597 |
-
resources = []
|
598 |
-
for score, doc_id, text in scored_documents[:10]:
|
599 |
-
doc_info = data['df'][data['df']['id'] == doc_id]
|
600 |
-
if doc_info.empty:
|
601 |
-
continue # Skip missing documents
|
602 |
-
doc_info = doc_info.iloc[0] # Safe to access the first row now
|
603 |
-
resources.append({
|
604 |
-
"id": doc_id,
|
605 |
-
"title": doc_info['title'],
|
606 |
-
"content": text[:200],
|
607 |
-
"score": float(score)
|
608 |
-
})
|
609 |
-
return {"resources": resources, "success": True}
|
610 |
|
611 |
|
612 |
@app.post("/api/recipes")
|
|
|
593 |
scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
|
594 |
scored_documents = list(zip(scores, document_ids, document_texts))
|
595 |
scored_documents.sort(key=lambda x: x[0], reverse=True)
|
596 |
+
return {"resources": scored_documents, "success": True}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
597 |
|
598 |
|
599 |
@app.post("/api/recipes")
|