Spaces:

umichVision
/

virtex-redcaps

Runtime error

zamborg commited on Nov 15, 2021

Commit

16484d3

•

1 Parent(s): b80df5c

syntax fix

Files changed (1) hide show

model.py CHANGED Viewed

@@ -58,7 +58,7 @@ class VirTexModel():
             subreddit_tokens = (
                 [self.model.sos_index] +
                 self.tokenizer.encode(subreddit_tokens) +
-                [tokenizer.token_to_id("[SEP]")]
                                )

             subreddit_tokens = (
                 [self.model.sos_index] +
                 self.tokenizer.encode(subreddit_tokens) +
+                [self.tokenizer.token_to_id("[SEP]")]
                                )