Srastog commited on
Commit
291adb4
·
verified ·
1 Parent(s): 97ce236

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +2 -1
app/main.py CHANGED
@@ -42,8 +42,9 @@ def upload_csv(uploaded_file:UploadFile=File(...)):
42
  @app.post("/upload/")
43
  async def upload_csv(file:UploadFile=File(...)):
44
  df=pd.read_csv(file.file)
 
45
  dataset=Dataset.from_pandas(df)
46
- return {"message": "CSV file uploaded and saved to Hugging Face Datasets"}
47
 
48
  @app.post("/train/")
49
  def training():
 
42
  @app.post("/upload/")
43
  async def upload_csv(file:UploadFile=File(...)):
44
  df=pd.read_csv(file.file)
45
+ return {"message": "CSV file uploaded and saved to Hugging Face Datasets 0"}
46
  dataset=Dataset.from_pandas(df)
47
+ return {"message": "CSV file uploaded and saved to Hugging Face Datasets 1"}
48
 
49
  @app.post("/train/")
50
  def training():