Spaces:

mridulk
/

phylo-diffusion

Sleeping

mridulk commited on Jun 21, 2024

Commit

f83e924

verified ·

1 Parent(s): 9d86095

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -89,7 +89,7 @@ def generate_image(fish_name, masking_level_input,
     all_images = []
     labels = []
-    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)

     all_images = []
     labels = []
+    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)