Update app/main.py

app/main.py

@@ -3,6 +3,7 @@ import glob
 import os
 
 import numpy as np
+import pandas
 from fastapi import FastAPI,UploadFile,File
 from pydantic import BaseModel,Field
 
@@ -23,6 +24,7 @@ app=FastAPI()
 def home():
     return {"message":"Hello World!"}
 
+"""
 @app.post("/upload/")
 def upload_csv(uploaded_file:UploadFile=File(...)):
     cwd=os.getcwd()
@@ -33,6 +35,14 @@ def upload_csv(uploaded_file:UploadFile=File(...)):
     return {'file': uploaded_file.filename,
             'content': uploaded_file.content_type,
             'path': path}
+"""
+
+@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")
+    return {"message": "CSV file uploaded and saved to Hugging Face Datasets"}
 
 @app.post("/train/")
 def training():