Srastog commited on
Commit
d044e51
·
verified ·
1 Parent(s): 088d766

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +8 -1
app/main.py CHANGED
@@ -11,6 +11,9 @@ from app.modelling import train
11
  from app.inference import predict
12
 
13
  dataset=None
 
 
 
14
 
15
  class Item(BaseModel):
16
  Torque:float=Field(gt=0,default=24.25)
@@ -36,7 +39,11 @@ def upload_csv(file:UploadFile=File(...)):
36
  @app.post("/train/")
37
  def training():
38
  results=train(dataset)
39
- return results
 
 
 
 
40
  #if dataset is not None:
41
  #else:
42
  # return {"message":"First Upload Dataset"}
 
11
  from app.inference import predict
12
 
13
  dataset=None
14
+ trained_model=None
15
+ encoder=None
16
+ transform=None
17
 
18
  class Item(BaseModel):
19
  Torque:float=Field(gt=0,default=24.25)
 
39
  @app.post("/train/")
40
  def training():
41
  results=train(dataset)
42
+ trained_model=results["model"]
43
+ encoder=results["encoder"]
44
+ transform=results["transform"]
45
+ return {"Accuracy":results["Accuracy"],
46
+ "F1_Score":results["F1_Score"]}
47
  #if dataset is not None:
48
  #else:
49
  # return {"message":"First Upload Dataset"}