Update app/main.py

app/main.py

@@ -10,6 +10,8 @@ from pydantic import BaseModel,Field
 from app.modelling import train
 from app.inference import predict
 
+dataset=[]
+
 class Item(BaseModel):
     Torque:float=Field(gt=0,default=24.25)
     Hydraulic_Pressure:float=Field(gt=0,default=121.86)
@@ -40,16 +42,13 @@ def upload_csv(uploaded_file:UploadFile=File(...)):
 @app.post("/upload/")
 async def upload_csv(file:UploadFile=File(...)):
     df=pd.read_csv(file.file)
-    dataset = Dataset.from_pandas(df)
-    dataset.push_to_hub("Srastog/app/data.csv")
+    dataset=Dataset.from_pandas(df)
     return {"message": "CSV file uploaded and saved to Hugging Face Datasets"}
 
 @app.post("/train/")
 def training():
-    cwd=os.getcwd()
-    path=os.path.join(cwd,"app","dataset.csv")
-    if os.path.exists(path):
-        results=train(path)
+    if len(dataset)>0:
+        results=train(dataset)
         return results
     else:
         return {"message":"First Upload Dataset"}