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

Update app/main.py

Browse files
Files changed (1) hide show
  1. 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 = 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():
49
- cwd=os.getcwd()
50
- path=os.path.join(cwd,"app","dataset.csv")
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"}