Update app.py

app.py

@@ -48,8 +48,8 @@ def masking_embed(embedding, levels=1):
 
 
 # LOAD MODEL GLOBALLY
-ckpt_path = '/globalscratch/mridul/ldm/final_runs_eccv/fishes/2024-03-01T23-15-36_HLE_days3/checkpoints/epoch=000119.ckpt'
-config_path = '/globalscratch/mridul/ldm/final_runs_eccv/fishes/2024-03-01T23-15-36_HLE_days3/configs/2024-03-01T23-15-36-project.yaml'
+ckpt_path = 'model_files/fishes/epoch=000119.ckpt'
+config_path = 'model_files/fishes/2024-03-01T23-15-36-project.yaml'
 config = OmegaConf.load(config_path)  # TODO: Optionally download from same location as ckpt and chnage this logic
 model = load_model_from_config(config, ckpt_path)  # TODO: check path
 
@@ -89,7 +89,7 @@ def generate_image(fish_name, masking_level_input,
     all_images = []
     labels = []
 
-    class_to_node = '/fastscratch/mridul/fishes/class_to_ancestral_label.pkl'
+    class_to_node = 'model_files/fishes/class_to_ancestral_label.pkl'
     with open(class_to_node, 'rb') as pickle_file:
         class_to_node_dict = pickle.load(pickle_file)