dsk129 commited on
Commit
352d3cc
·
verified ·
1 Parent(s): 8f25d3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def classify_residues(seq):
73
 
74
  def compute_cosine_heatmap(seq):
75
  #--tokenize
76
- inputs = tokenizer(seq, reuturn_tenors="pt")
77
  with torch.no_grad():
78
  outputs = model(**inputs)
79
  embedding = outputs.last_hidden_state[0]
 
73
 
74
  def compute_cosine_heatmap(seq):
75
  #--tokenize
76
+ inputs = tokenizer(seq, return_tenors="pt")
77
  with torch.no_grad():
78
  outputs = model(**inputs)
79
  embedding = outputs.last_hidden_state[0]