dayannex commited on
Commit
d2105df
·
1 Parent(s): a03f45e

correcion modelo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -220,7 +220,7 @@ class Model:
220
 
221
  inputs = self.tokenizer(self.texto, return_tensors="pt")
222
  with torch.no_grad():
223
- outputs = model(**inputs)
224
  logits = outputs.logits
225
  predictions = torch.argmax(logits, dim=2)
226
 
 
220
 
221
  inputs = self.tokenizer(self.texto, return_tensors="pt")
222
  with torch.no_grad():
223
+ outputs = self.model(**inputs)
224
  logits = outputs.logits
225
  predictions = torch.argmax(logits, dim=2)
226