nataliegref commited on
Commit
3e82392
·
1 Parent(s): 4aa3e15
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -3,6 +3,7 @@ from pydantic import BaseModel
3
  import numpy as np
4
  import pickle
5
  from huggingface_hub import hf_hub_download
 
6
 
7
  app = FastAPI()
8
 
@@ -13,9 +14,9 @@ with open(model_path, "rb") as f:
13
  model = pickle.load(f)
14
 
15
  class InputData(BaseModel):
16
- X: list
17
- T0: list | None = None
18
- T1: list | None = None
19
 
20
  @app.post("/model-effect")
21
  def predict(data: InputData):
 
3
  import numpy as np
4
  import pickle
5
  from huggingface_hub import hf_hub_download
6
+ from typing import List, Optional
7
 
8
  app = FastAPI()
9
 
 
14
  model = pickle.load(f)
15
 
16
  class InputData(BaseModel):
17
+ X: List
18
+ T0: Optional[List] = None
19
+ T1: Optional[List] = None
20
 
21
  @app.post("/model-effect")
22
  def predict(data: InputData):