Update app.py

app.py

@@ -285,6 +285,10 @@ class Engine(object):
         # Load the CSV
         df = pd.read_csv(csv_file)
 
+        df['id'] = df['id'].astype(str)
+        df = pd.concat([df[df['id'] == '495204901603170'], df[df['id'] != '495204901603170']])
+        df = pd.concat([df[df['id'] == '732681614433401'], df[df['id'] != '732681614433401']])
+
         # order
         self.order = np.arange(len(df))
         np.random.shuffle(self.order)  # Shuffle the order array
@@ -293,9 +297,6 @@ class Engine(object):
         df = df.iloc[self.order].reset_index(drop=True)
 
         # Put image with id 732681614433401 on the top and then all the rest below
-        df['id'] = df['id'].astype(str)
-        #df = pd.concat([df[df['id'] == '495204901603170'], df[df['id'] != '495204901603170']])
-        #df = pd.concat([df[df['id'] == '732681614433401'], df[df['id'] != '732681614433401']])
 
         # Get the image filenames and their coordinates
         self.images = [os.path.join(self.image_folder, f"{img_path}.jpg") for img_path in df['id'].tolist()[:]]