god i dummy pt4

model.py

@@ -63,7 +63,7 @@ class VirTexModel():
             subreddit_tokens = torch.tensor(subreddit_tokens, device=self.device).long()
             
         if prompt is not "":
-            cap_tokens = self.tokenizer.encode(subreddit_tokens) 
+            cap_tokens = self.tokenizer.encode(prompt) 
             subreddit_tokens = torch.cat((subreddit_tokens, cap_tokens))
             
         predictions: List[Dict[str, Any]] = []