Srastog commited on
Commit
17f0ebe
·
verified ·
1 Parent(s): f5d939c

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +5 -5
app/main.py CHANGED
@@ -35,11 +35,11 @@ def upload_csv(file:UploadFile=File(...)):
35
 
36
  @app.post("/train/")
37
  def training():
38
- if dataset is not None:
39
- results=train(dataset)
40
- return results
41
- else:
42
- return {"message":"First Upload Dataset"}
43
 
44
  @app.post("/predict/")
45
  def prediction(item:Item):
 
35
 
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"}
43
 
44
  @app.post("/predict/")
45
  def prediction(item:Item):