fixing st writeouts

app.py

@@ -7,6 +7,8 @@ sys.path.append("./virtex/")
 
 def gen_show_caption(sub_prompt=None, cap_prompt = ""):
     with st.spinner("Generating Caption"):
+        if sub_prompt is None and cap_prompt is not "":
+            st.write("Without a specified subreddit, caption prompts will skip subreddit prediction")
         subreddit, caption = virtexModel.predict(image_dict, sub_prompt=sub_prompt, prompt = cap_prompt)
         st.header("Predicted Caption:\n\n")
         st.subheader(f"r/{subreddit}:\t{caption}\n")

model.py

@@ -63,13 +63,12 @@ class VirTexModel():
             subreddit_tokens = torch.tensor(subreddit_tokens, device=self.device).long()
             
         if prompt is not "":
+            # at present prompts without subreddits will break without this change
+            # TODO FIX
             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]] = []