eaedk commited on
Commit
5bd316b
1 Parent(s): a82f936

cleaning, api desc added

Browse files
Files changed (1) hide show
  1. main.py +7 -12
main.py CHANGED
@@ -8,16 +8,9 @@ import os
8
  import json
9
  import logging
10
 
11
- # logger = logging.getLogger(__name__)
12
  logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.DEBUG)
13
 
14
- # logging.debug('This is a debug message')
15
- # logging.info('This is an info message')
16
- # logging.warning('This is a warning message')
17
- # logging.error('This is an error message')
18
- # logging.critical('This is a critical message')
19
-
20
-
21
  # Util Functions & Classes
22
  def loading(fp):
23
  with open(fp, "rb") as f:
@@ -102,20 +95,22 @@ class Lands(BaseModel):
102
  return [i.dict() for i in cls.inputs]
103
 
104
 
105
- # API
106
- app = FastAPI(title="API")
 
 
 
107
  ml_objects = loading(fp=os.path.join("assets", "ml", "crop_recommandation2.pkl"))
108
  ## Extract the ml components
109
  model = ml_objects["model"]
110
  scaler = ml_objects["scaler"].set_output(transform="pandas")
111
  labels = ml_objects["labels"]
112
- # = ml_objects[""]
113
 
114
 
115
  # Endpoints
116
  @app.get("/")
117
  def root():
118
- return {"API": " This is an API to ... ."}
119
 
120
 
121
  @app.get("/checkup")
 
8
  import json
9
  import logging
10
 
11
+ # logger
12
  logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.DEBUG)
13
 
 
 
 
 
 
 
 
14
  # Util Functions & Classes
15
  def loading(fp):
16
  with open(fp, "rb") as f:
 
95
  return [i.dict() for i in cls.inputs]
96
 
97
 
98
+ # API Config
99
+ app = FastAPI(title="Agri-Tech API",
100
+ description="This is a ML API for classification of crop to plant on a land regarding some features")
101
+
102
+ # ML Config
103
  ml_objects = loading(fp=os.path.join("assets", "ml", "crop_recommandation2.pkl"))
104
  ## Extract the ml components
105
  model = ml_objects["model"]
106
  scaler = ml_objects["scaler"].set_output(transform="pandas")
107
  labels = ml_objects["labels"]
 
108
 
109
 
110
  # Endpoints
111
  @app.get("/")
112
  def root():
113
+ return {"Description": " This is a ML API for classification of crop to plant on a land regarding some features."}
114
 
115
 
116
  @app.get("/checkup")