data

app.py

@@ -49,6 +49,9 @@ class RangeModel(nn.Module):
 
 model = RangeModel()
 model.load_state_dict(torch.load("model/demo_model.pt", map_location=torch.device('cpu')))
+pos_embed = np.load("data/pos_embed.npy", allow_pickle=True)
+model.x = torch.tensor(pos_embed).unsqueeze(0).float()
+model.eval()
 
 species = np.load("data/species_70b.npy", allow_pickle=True)
 clas = np.load("data/class_70b.npy", allow_pickle=True)
@@ -62,8 +65,6 @@ order_list = list(order[()].keys())
 #genus_list = list(genus[()].keys())
 #family_list = list(family[()].keys())
 
-#pos_embed = np.load("data/pos_embed.npy", allow_pickle=True)
-
 def update_fn(val):
     if val=="Class":
         return gr.Dropdown(label="Name", choices=class_list, interactive=True)