Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
@@ -10,7 +10,7 @@ current_dir = Path(__file__).parent
|
|
10 |
|
11 |
# Load the model
|
12 |
fhe_model = FHEModelServer("deployment/financial_rating")
|
13 |
-
fhe_legal_model = FHEModelServer("deployment/legal_rating")
|
14 |
class PredictRequest(BaseModel):
|
15 |
evaluation_key: str
|
16 |
encrypted_encoding: str
|
@@ -25,7 +25,7 @@ def root():
|
|
25 |
|
26 |
@app.post("/predict_sentiment")
|
27 |
def predict_sentiment(query: PredictRequest):
|
28 |
-
|
29 |
|
30 |
encrypted_encoding = base64.b64decode(query.encrypted_encoding)
|
31 |
evaluation_key = base64.b64decode(query.evaluation_key)
|
|
|
10 |
|
11 |
# Load the model
|
12 |
fhe_model = FHEModelServer("deployment/financial_rating")
|
13 |
+
#fhe_legal_model = FHEModelServer("deployment/legal_rating")
|
14 |
class PredictRequest(BaseModel):
|
15 |
evaluation_key: str
|
16 |
encrypted_encoding: str
|
|
|
25 |
|
26 |
@app.post("/predict_sentiment")
|
27 |
def predict_sentiment(query: PredictRequest):
|
28 |
+
|
29 |
|
30 |
encrypted_encoding = base64.b64decode(query.encrypted_encoding)
|
31 |
evaluation_key = base64.b64decode(query.evaluation_key)
|