data

app.py

@@ -50,7 +50,7 @@ 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_embeds_model.npy", allow_pickle=True)
-model.x = torch.tensor(pos_embed).unsqueeze(0).float()
+model.x = torch.tensor(pos_embed).float()
 model.eval()
 
 species = np.load("data/species_70b.npy", allow_pickle=True)