Spaces:

mudaza
/

semantic-ir-diagnose-disease

Sleeping

mudaza commited on Aug 28, 2024

Commit

c72cbc0

1 Parent(s): 59eb01b

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import pickle
 import pandas as pd
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
 corpus = pickle.load(open("./corpus/all_embeddings.pickle", "rb"))
 label_encoder = pickle.load(open("./corpus/label_encoder.pickle", "rb"))

 import pandas as pd
 from pydantic import BaseModel
 from fastapi.middleware.cors import CORSMiddleware
+import torch
 corpus = pickle.load(open("./corpus/all_embeddings.pickle", "rb"))
 label_encoder = pickle.load(open("./corpus/label_encoder.pickle", "rb"))