Srastog commited on
Commit
02347de
·
verified ·
1 Parent(s): 291adb4

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +3 -2
app/main.py CHANGED
@@ -42,9 +42,10 @@ 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
- 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():
 
42
  @app.post("/upload/")
43
  async def upload_csv(file:UploadFile=File(...)):
44
  df=pd.read_csv(file.file)
45
+ global dataset
46
  dataset=Dataset.from_pandas(df)
47
+ file.file.close()
48
+ return {"filename": file.filename}
49
 
50
  @app.post("/train/")
51
  def training():