Spaces:

SERHANI
/

model2

Sleeping

SERHANI commited on Nov 12, 2024

Commit

c4b0dac

verified ·

1 Parent(s): bf81a2d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from PIL import Image, ImageDraw
 import io
 app = Flask(__name__)
-model = YOLO('last.pt')  # Load your YOLO model
 class_names = ['Acne', 'Dark circles', 'blackheads', 'eczema', 'rosacea', 'whiteheads', 'wrinkles']
 @app.route('/classify', methods=['POST'])

 import io
 app = Flask(__name__)
+model = YOLO('best.pt')  # Load your YOLO model
 class_names = ['Acne', 'Dark circles', 'blackheads', 'eczema', 'rosacea', 'whiteheads', 'wrinkles']
 @app.route('/classify', methods=['POST'])