update code
Browse files- app.py +1 -0
- 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
|