Update README.md

README.md

@@ -96,7 +96,7 @@ probabilities = torch.nn.functional.softmax(logits, dim=-1)
 print(f"Probabilities shape: {probabilities.shape}")
 
 # Get probabilities associated with intron
-idx_intron = features.index("intron")
+idx_intron = model.config.features.index("intron")
 probabilities_intron = probabilities[:,:,idx_intron]
 print(f"Intron probabilities shape: {probabilities_intron.shape}")