Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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', '
|
| 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 = ['Circle', 'Square', 'Triangle', 'Rectangle']
|
| 15 |
|
| 16 |
def predict(image):
|
| 17 |
response = {}
|