modified code and add files
Browse files
app.py
CHANGED
@@ -33,6 +33,7 @@ class Symptoms(BaseModel):
|
|
33 |
|
34 |
@app.get("/")
|
35 |
def home():
|
|
|
36 |
return {"Hello": "World!"}
|
37 |
|
38 |
@app.post("/", response_model=list[Disease])
|
@@ -42,6 +43,7 @@ async def predict(symptoms: Symptoms):
|
|
42 |
scores, indicies = torch.topk(similarity_vectors, k=len(corpus))
|
43 |
# id_ = df.iloc[indicies].reset_index(drop=True)
|
44 |
ls = df.iloc[indicies].copy()
|
|
|
45 |
# id_ = id_.drop_duplicates("label")
|
46 |
ls["scores"] = scores
|
47 |
# scores = scores[id_.index]
|
@@ -51,5 +53,5 @@ async def predict(symptoms: Symptoms):
|
|
51 |
"name": value[1],
|
52 |
"url" : value[2],
|
53 |
"score" : value[3]})
|
54 |
-
for value in zip(
|
55 |
return diseases
|
|
|
33 |
|
34 |
@app.get("/")
|
35 |
def home():
|
36 |
+
print(df.iloc[0])
|
37 |
return {"Hello": "World!"}
|
38 |
|
39 |
@app.post("/", response_model=list[Disease])
|
|
|
43 |
scores, indicies = torch.topk(similarity_vectors, k=len(corpus))
|
44 |
# id_ = df.iloc[indicies].reset_index(drop=True)
|
45 |
ls = df.iloc[indicies].copy()
|
46 |
+
print(ls.iloc[0])
|
47 |
# id_ = id_.drop_duplicates("label")
|
48 |
ls["scores"] = scores
|
49 |
# scores = scores[id_.index]
|
|
|
53 |
"name": value[1],
|
54 |
"url" : value[2],
|
55 |
"score" : value[3]})
|
56 |
+
for value in zip(ls.index, ls["name"], ls["url"], ls["scores"])]
|
57 |
return diseases
|