12 lines
295 B
Python
12 lines
295 B
Python
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
class Config:
|
||
|
|
UPLOAD_FOLDER = 'static/uploads'
|
||
|
|
RESULT_FOLDER = 'static/results'
|
||
|
|
HARDCODED_IMAGES = 'training/memory'
|
||
|
|
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
|
||
|
|
|
||
|
|
# Model configuration
|
||
|
|
MODEL_PATH = 'models/memory_detector.pt'
|
||
|
|
CONFIDENCE_THRESHOLD = 0.5
|