testing vicuna output with prints

app.py

@@ -60,8 +60,11 @@ def vicuna_respond(tab_name, message, chat_history):
     print('Vicuna - Prompt + Context:')
     print(formatted_prompt)
     input_ids = vicuna_tokenizer.encode(formatted_prompt, return_tensors="pt")
+    print(input_ids)
     output_ids = vicuna_model.generate(input_ids, do_sample=True, max_length=1024, num_beams=5, no_repeat_ngram_size=2)
+    print(output_ids)
     bot_message = vicuna_tokenizer.decode(output_ids[0], skip_special_tokens=True)
+    print(bot_message)
     # Remove formatted prompt from bot_message
     bot_message = bot_message.replace(formatted_prompt, '')
     print(bot_message)
@@ -116,11 +119,11 @@ def interface():
         #     gpt_S3_chatbot = gr.Chatbot(label="gpt-3.5")
         # clear = gr.ClearButton(components=[textbox_prompt, vicuna_S3_chatbot])
 
-        textbox_prompt.submit(vicuna_respond, inputs=[textbox_prompt, vicuna_S1_chatbot], outputs=[textbox_prompt, vicuna_S1_chatbot])
+        #textbox_prompt.submit(vicuna_respond, inputs=[textbox_prompt, vicuna_S1_chatbot], outputs=[textbox_prompt, vicuna_S1_chatbot])
         # textbox_prompt.submit(respond, inputs=[textbox_prompt, vicuna_S2_chatbot], outputs=[textbox_prompt, vicuna_S2_chatbot])
         # textbox_prompt.submit(respond, inputs=[textbox_prompt, vicuna_S3_chatbot], outputs=[textbox_prompt, vicuna_S3_chatbot])
 
-        textbox_prompt.submit(llama_respond, inputs=[textbox_prompt, llama_S1_chatbot], outputs=[textbox_prompt, llama_S1_chatbot])
+        #textbox_prompt.submit(llama_respond, inputs=[textbox_prompt, llama_S1_chatbot], outputs=[textbox_prompt, llama_S1_chatbot])
 
         btn.click(vicuna_respond, inputs=[tab_name, textbox_prompt, vicuna_S1_chatbot], outputs=[tab_name, textbox_prompt, vicuna_S1_chatbot])
         btn.click(llama_respond, inputs=[tab_name, textbox_prompt, llama_S1_chatbot], outputs=[tab_name, textbox_prompt, llama_S1_chatbot])