prompting with no subreddit

model.py

@@ -63,9 +63,14 @@ class VirTexModel():
             subreddit_tokens = torch.tensor(subreddit_tokens, device=self.device).long()
             
         if prompt is not "":
-            cap_tokens = self.tokenizer.encode(prompt)
-            cap_tokens = torch.tensor(cap_tokens, device=self.device).long()
-            subreddit_tokens = torch.cat([subreddit_tokens, cap_tokens])
+            if sub_prompt is not None:
+                cap_tokens = self.tokenizer.encode(prompt)
+                cap_tokens = torch.tensor(cap_tokens, device=self.device).long()
+                subreddit_tokens = torch.cat([subreddit_tokens, cap_tokens])
+            else:
+                st.write("Without a specified subreddit, caption prompts will skip subreddit prediction")
+                #TODO fix this
+            
             
         predictions: List[Dict[str, Any]] = []
         
@@ -90,7 +95,7 @@ class VirTexModel():
             else:
                 subreddit, rest_of_caption = "", caption
             
-            is_valid_subreddit = True if sub_prompt is not None else subreddit in self.valid_subs
+            is_valid_subreddit = True if sub_prompt is not None or prompt is not None else subreddit in self.valid_subs
             
             
         return subreddit, rest_of_caption