yinuozhang commited on
Commit
7c0621c
·
verified ·
1 Parent(s): d76f5aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model = AutoModel.from_pretrained("ChatterjeeLab/MetaLATTE", config=config)
17
 
18
  def predict(sequence):
19
  inputs = tokenizer(sequence, return_tensors="pt")
20
- raw_probs, predictions = model(**inputs)
21
 
22
  id2label = config.id2label
23
  results = {}
 
17
 
18
  def predict(sequence):
19
  inputs = tokenizer(sequence, return_tensors="pt")
20
+ raw_probs, predictions = model.predict(**inputs)
21
 
22
  id2label = config.id2label
23
  results = {}