Update app/main.py

app/main.py

@@ -11,6 +11,9 @@ from app.modelling import train
 from app.inference import predict
 
 dataset=None
+trained_model=None
+encoder=None
+transform=None
 
 class Item(BaseModel):
     Torque:float=Field(gt=0,default=24.25)
@@ -36,7 +39,11 @@ def upload_csv(file:UploadFile=File(...)):
 @app.post("/train/")
 def training():
     results=train(dataset)
-    return results
+    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"}