msabia commited on
Commit
46bdc00
·
verified ·
1 Parent(s): 8a6d01d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
app.py CHANGED
@@ -8,7 +8,16 @@ import io
8
  import pickle
9
  import random
10
 
11
- # add random index retreival
 
 
 
 
 
 
 
 
 
12
 
13
  def get_image(image1, image2, dataset_image_mask, processid_to_index, idx):
14
  if (idx < 162834):
@@ -75,17 +84,6 @@ def searchEmbeddings(id, mod1, mod2):
75
 
76
  return id_list, image0, image1, image2, image3, image4, image5, image6, image7, image8, image9, image10, s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10
77
 
78
- def getRandID():
79
- indx = random.randrange(0, 325667)
80
- return indx_to_id_dict[indx]
81
-
82
- def getTax(indx):
83
- s = species[indx]
84
- g = genus[indx]
85
- f = family[indx]
86
- str = "Species: " + s + "\nGenus: " + g + "\nFamily: " + f
87
- return str
88
-
89
  with gr.Blocks() as demo:
90
  # open general files
91
  with open("dataset_image1.pickle", "rb") as f:
 
8
  import pickle
9
  import random
10
 
11
+ def getRandID():
12
+ indx = random.randrange(0, 325667)
13
+ return indx_to_id_dict[indx]
14
+
15
+ def getTax(indx):
16
+ s = species[indx]
17
+ g = genus[indx]
18
+ f = family[indx]
19
+ str = "Species: " + s + "\nGenus: " + g + "\nFamily: " + f
20
+ return str
21
 
22
  def get_image(image1, image2, dataset_image_mask, processid_to_index, idx):
23
  if (idx < 162834):
 
84
 
85
  return id_list, image0, image1, image2, image3, image4, image5, image6, image7, image8, image9, image10, s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10
86
 
 
 
 
 
 
 
 
 
 
 
 
87
  with gr.Blocks() as demo:
88
  # open general files
89
  with open("dataset_image1.pickle", "rb") as f: