mudaza commited on
Commit
8f2a805
·
1 Parent(s): 0bfc10b

modified code and add files

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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(df.index, ls["name"], ls["url"], ls["scores"])]
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