Hugging Face's logo

ChatterjeeLab
/
MetaLATTE

Transformers
PyTorch
metalatte
climate
biology
Model card Files Community
yinuozhang commited on
Commit
d749383
·
1 Parent(s): 071db43

update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ model = AutoModel.from_pretrained("ChatterjeeLab/MetaLATTE", config=config)
40
  model.eval()
41
  sequence = "AVYNIGWSFNVNGARGKSFRAGDVLVFKYIKGQHNVVAVNGRGYASCSAPRGARTYSSGQDRIKLTRGQNYFICSFPGHCGGGMKIAINAK"
42
  inputs = tokenizer(sequence, return_tensors="pt")
43
- raw_probs, predictions = model(**inputs)
44
 
45
  id2label = config.id2label
46
  predicted_labels = [id2label[i] for i, pred in enumerate(predictions[0]) if pred == 1]
 
40
  model.eval()
41
  sequence = "AVYNIGWSFNVNGARGKSFRAGDVLVFKYIKGQHNVVAVNGRGYASCSAPRGARTYSSGQDRIKLTRGQNYFICSFPGHCGGGMKIAINAK"
42
  inputs = tokenizer(sequence, return_tensors="pt")
43
+ raw_probs, predictions = model.predict(**inputs)
44
 
45
  id2label = config.id2label
46
  predicted_labels = [id2label[i] for i, pred in enumerate(predictions[0]) if pred == 1]