Srastog commited on
Commit
14b8a20
·
verified ·
1 Parent(s): cba7086

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +10 -0
app/main.py CHANGED
@@ -3,6 +3,7 @@ import glob
3
  import os
4
 
5
  import numpy as np
 
6
  from fastapi import FastAPI,UploadFile,File
7
  from pydantic import BaseModel,Field
8
 
@@ -23,6 +24,7 @@ app=FastAPI()
23
  def home():
24
  return {"message":"Hello World!"}
25
 
 
26
  @app.post("/upload/")
27
  def upload_csv(uploaded_file:UploadFile=File(...)):
28
  cwd=os.getcwd()
@@ -33,6 +35,14 @@ def upload_csv(uploaded_file:UploadFile=File(...)):
33
  return {'file': uploaded_file.filename,
34
  'content': uploaded_file.content_type,
35
  'path': path}
 
 
 
 
 
 
 
 
36
 
37
  @app.post("/train/")
38
  def training():
 
3
  import os
4
 
5
  import numpy as np
6
+ import pandas
7
  from fastapi import FastAPI,UploadFile,File
8
  from pydantic import BaseModel,Field
9
 
 
24
  def home():
25
  return {"message":"Hello World!"}
26
 
27
+ """
28
  @app.post("/upload/")
29
  def upload_csv(uploaded_file:UploadFile=File(...)):
30
  cwd=os.getcwd()
 
35
  return {'file': uploaded_file.filename,
36
  'content': uploaded_file.content_type,
37
  'path': path}
38
+ """
39
+
40
+ @app.post("/upload/")
41
+ async def upload_csv(file:UploadFile=File(...)):
42
+ df=pd.read_csv(file.file)
43
+ dataset = Dataset.from_pandas(df)
44
+ dataset.push_to_hub("Srastog/app/data.csv")
45
+ return {"message": "CSV file uploaded and saved to Hugging Face Datasets"}
46
 
47
  @app.post("/train/")
48
  def training():