Update app/main.py

app/main.py

@@ -35,11 +35,11 @@ def upload_csv(file:UploadFile=File(...)):
 
 @app.post("/train/")
 def training():
-    if dataset is not None:
-        results=train(dataset)
-        return results
-    else:
-        return {"message":"First Upload Dataset"}
+    results=train(dataset)
+    return results
+    #if dataset is not None:
+    #else:
+    #    return {"message":"First Upload Dataset"}
 
 @app.post("/predict/")
 def prediction(item:Item):