Srastog commited on
Commit
11b447b
·
verified ·
1 Parent(s): d044e51

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +9 -10
app/main.py CHANGED
@@ -31,22 +31,21 @@ def home():
31
 
32
  @app.post("/upload/")
33
  def upload_csv(file:UploadFile=File(...)):
34
- global dataset
35
  dataset=pd.read_csv(file.file)
36
  file.file.close()
37
  return {"filename": file.filename}
38
 
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"}
50
 
51
  @app.post("/predict/")
52
  def prediction(item:Item):
 
31
 
32
  @app.post("/upload/")
33
  def upload_csv(file:UploadFile=File(...)):
 
34
  dataset=pd.read_csv(file.file)
35
  file.file.close()
36
  return {"filename": file.filename}
37
 
38
  @app.post("/train/")
39
  def training():
40
+ if dataset is not None:
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
+ else:
48
+ return {"message":"First Upload Dataset"}
49
 
50
  @app.post("/predict/")
51
  def prediction(item:Item):