Srastog commited on
Commit
ef73d9b
·
verified ·
1 Parent(s): 1a8c89e

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +10 -5
app/main.py CHANGED
@@ -14,7 +14,6 @@ dataset=None
14
  trained_model=None
15
  encoder=None
16
  transform=None
17
- step=0
18
 
19
  class Item(BaseModel):
20
  Torque:float=Field(gt=0,default=24.25)
@@ -32,19 +31,25 @@ def home():
32
 
33
  @app.post("/upload/")
34
  def upload_csv(file:UploadFile=File(...)):
35
- step=1
36
  dataset=pd.read_csv(file.file)
37
  file.file.close()
38
  return {"filename": file.filename}
39
 
40
  @app.post("/train/")
41
  def training():
42
- if step>0:
43
- step=2
44
  results=train(dataset)
 
 
45
  trained_model=results["model"]
 
 
46
  encoder=results["encoder"]
 
 
47
  transform=results["transform"]
 
48
  return {"Accuracy":results["Accuracy"],
49
  "F1_Score":results["F1_Score"]}
50
  else:
@@ -52,7 +57,7 @@ def training():
52
 
53
  @app.post("/predict/")
54
  def prediction(item:Item):
55
- if step>1:
56
  arr=[[item.Torque,item.Hydraulic_Pressure,item.Cutting,item.Coolant_Pressure,item.Spindle_Speed,item.Coolant_Temperature]]
57
  results=predict(trained_model,encoder,transform,arr)
58
  return results
 
14
  trained_model=None
15
  encoder=None
16
  transform=None
 
17
 
18
  class Item(BaseModel):
19
  Torque:float=Field(gt=0,default=24.25)
 
31
 
32
  @app.post("/upload/")
33
  def upload_csv(file:UploadFile=File(...)):
34
+ global dataset
35
  dataset=pd.read_csv(file.file)
36
  file.file.close()
37
  return {"filename": file.filename}
38
 
39
  @app.post("/train/")
40
  def training():
41
+ if dataset is not None:
 
42
  results=train(dataset)
43
+
44
+ global trained_model
45
  trained_model=results["model"]
46
+
47
+ global encoder
48
  encoder=results["encoder"]
49
+
50
+ global transform
51
  transform=results["transform"]
52
+
53
  return {"Accuracy":results["Accuracy"],
54
  "F1_Score":results["F1_Score"]}
55
  else:
 
57
 
58
  @app.post("/predict/")
59
  def prediction(item:Item):
60
+ if trained_model is not None:
61
  arr=[[item.Torque,item.Hydraulic_Pressure,item.Cutting,item.Coolant_Pressure,item.Spindle_Speed,item.Coolant_Temperature]]
62
  results=predict(trained_model,encoder,transform,arr)
63
  return results