Artisteve commited on
Commit
bb89631
·
1 Parent(s): f4fd059

reverting changes to the last working version

Browse files
Files changed (2) hide show
  1. __pycache__/main.cpython-310.pyc +0 -0
  2. main.py +2 -2
__pycache__/main.cpython-310.pyc ADDED
Binary file (1.37 kB). View file
 
main.py CHANGED
@@ -9,8 +9,8 @@ app = FastAPI()
9
 
10
  # Define paths
11
  UPLOAD_FOLDER = "uploaded_images"
12
- MODEL_FILENAME = "model_1000.pkl"
13
- CSV_FILE = "metadata_filtered_1000.csv"
14
 
15
  # Ensure the upload folder exists
16
  os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
9
 
10
  # Define paths
11
  UPLOAD_FOLDER = "uploaded_images"
12
+ MODEL_FILENAME = "kmeans_model_100.pkl"
13
+ CSV_FILE = "cluster_assignments_100.csv"
14
 
15
  # Ensure the upload folder exists
16
  os.makedirs(UPLOAD_FOLDER, exist_ok=True)