Update app/main.py

app/main.py

@@ -14,7 +14,6 @@ dataset=None
 trained_model=None
 encoder=None
 transform=None
-step=0
 
 class Item(BaseModel):
     Torque:float=Field(gt=0,default=24.25)
@@ -32,19 +31,25 @@ def home():
 
 @app.post("/upload/")
 def upload_csv(file:UploadFile=File(...)):
-    step=1
+    global dataset
     dataset=pd.read_csv(file.file)
     file.file.close()
     return {"filename": file.filename}
 
 @app.post("/train/")
 def training():
-    if step>0:
-        step=2
+    if dataset is not None:
         results=train(dataset)
+        
+        global trained_model
         trained_model=results["model"]
+        
+        global encoder
         encoder=results["encoder"]
+        
+        global transform
         transform=results["transform"]
+        
         return {"Accuracy":results["Accuracy"],
                 "F1_Score":results["F1_Score"]}
     else:
@@ -52,7 +57,7 @@ def training():
 
 @app.post("/predict/")
 def prediction(item:Item):
-    if step>1:
+    if trained_model is not None:
         arr=[[item.Torque,item.Hydraulic_Pressure,item.Cutting,item.Coolant_Pressure,item.Spindle_Speed,item.Coolant_Temperature]]
         results=predict(trained_model,encoder,transform,arr)
         return results