Update app/main.py

app/main.py

@@ -31,22 +31,21 @@ def home():
 
 @app.post("/upload/")
 def upload_csv(file:UploadFile=File(...)):
-    global dataset
     dataset=pd.read_csv(file.file)
     file.file.close()
     return {"filename": file.filename}
 
 @app.post("/train/")
 def training():
-    results=train(dataset)
-    trained_model=results["model"]
-    encoder=results["encoder"]
-    transform=results["transform"]
-    return {"Accuracy":results["Accuracy"],
-            "F1_Score":results["F1_Score"]}
-    #if dataset is not None:
-    #else:
-    #    return {"message":"First Upload Dataset"}
+    if dataset is not None:
+        results=train(dataset)
+        trained_model=results["model"]
+        encoder=results["encoder"]
+        transform=results["transform"]
+        return {"Accuracy":results["Accuracy"],
+                "F1_Score":results["F1_Score"]}
+    else:
+        return {"message":"First Upload Dataset"}
 
 @app.post("/predict/")
 def prediction(item:Item):