Update app/main.py

app/main.py

@@ -3,14 +3,14 @@ import glob
 import os
 
 import numpy as np
-import pandas
+import pandas as pd
 from fastapi import FastAPI,UploadFile,File
 from pydantic import BaseModel,Field
 
 from app.modelling import train
 from app.inference import predict
 
-dataset=[]
+dataset=None
 
 class Item(BaseModel):
     Torque:float=Field(gt=0,default=24.25)
@@ -49,7 +49,7 @@ async def upload_csv(file:UploadFile=File(...)):
 
 @app.post("/train/")
 def training():
-    if len(dataset)>0:
+    if dataset is not None:
         results=train(dataset)
         return results
     else: