Update app/main.py
Browse files- 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():
|