Update app/main.py
Browse files- app/main.py +5 -6
app/main.py
CHANGED
@@ -10,6 +10,8 @@ from pydantic import BaseModel,Field
|
|
10 |
from app.modelling import train
|
11 |
from app.inference import predict
|
12 |
|
|
|
|
|
13 |
class Item(BaseModel):
|
14 |
Torque:float=Field(gt=0,default=24.25)
|
15 |
Hydraulic_Pressure:float=Field(gt=0,default=121.86)
|
@@ -40,16 +42,13 @@ def upload_csv(uploaded_file:UploadFile=File(...)):
|
|
40 |
@app.post("/upload/")
|
41 |
async def upload_csv(file:UploadFile=File(...)):
|
42 |
df=pd.read_csv(file.file)
|
43 |
-
dataset
|
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():
|
49 |
-
|
50 |
-
|
51 |
-
if os.path.exists(path):
|
52 |
-
results=train(path)
|
53 |
return results
|
54 |
else:
|
55 |
return {"message":"First Upload Dataset"}
|
|
|
10 |
from app.modelling import train
|
11 |
from app.inference import predict
|
12 |
|
13 |
+
dataset=[]
|
14 |
+
|
15 |
class Item(BaseModel):
|
16 |
Torque:float=Field(gt=0,default=24.25)
|
17 |
Hydraulic_Pressure:float=Field(gt=0,default=121.86)
|
|
|
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():
|
50 |
+
if len(dataset)>0:
|
51 |
+
results=train(dataset)
|
|
|
|
|
52 |
return results
|
53 |
else:
|
54 |
return {"message":"First Upload Dataset"}
|