Laufey commited on
Commit
5f5ea10
·
verified ·
1 Parent(s): 0e03260

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ xgb_quality = joblib.load("xgb_quality.pkl")
11
  scaler = joblib.load("scaler.pkl")
12
  le_quality = joblib.load("le_quality.pkl")
13
  feature_extractor = GeometricFeatureExtractor()
14
- SHAPES = ['Circle', 'Square', 'Triangle', 'Rectangle']
15
 
16
  def predict(image):
17
  response = {}
 
11
  scaler = joblib.load("scaler.pkl")
12
  le_quality = joblib.load("le_quality.pkl")
13
  feature_extractor = GeometricFeatureExtractor()
14
+ SHAPES = ['Triangle', 'Square', 'Circle', 'Rectangle']
15
 
16
  def predict(image):
17
  response = {}