mudaza commited on
Commit
ade21fe
·
1 Parent(s): 874d7b1

update code

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ from fastapi import FastAPI
3
  import pickle
4
  import pandas as pd
5
  from pydantic import BaseModel
 
6
 
7
  corpus = pickle.load(open("./corpus/all_embeddings.pickle", "rb"))
8
  label_encoder = pickle.load(open("./corpus/label_encoder.pickle", "rb"))
 
3
  import pickle
4
  import pandas as pd
5
  from pydantic import BaseModel
6
+ from fastapi.middleware.cors import CORSMiddleware
7
 
8
  corpus = pickle.load(open("./corpus/all_embeddings.pickle", "rb"))
9
  label_encoder = pickle.load(open("./corpus/label_encoder.pickle", "rb"))
requirements.txt CHANGED
@@ -3,3 +3,4 @@ uvicorn[standard]
3
  pandas
4
  sentence-transformers
5
  pydantic
 
 
3
  pandas
4
  sentence-transformers
5
  pydantic
6
+ scikit-learn==1.3.2