added names to print

app.py

@@ -57,7 +57,7 @@ def gpt_respond(tab_name, message, chat_history, max_convo_length = 10):
 
 def vicuna_respond(tab_name, message, chat_history):
     formatted_prompt = f'''Generate the output only for the assistant. Please output any {tab_name} in the following sentence one per line without any additional text: {message}'''
-    print('Prompt + Context:')
+    print('Vicuna - Prompt + Context:')
     print(formatted_prompt)
     input_ids = vicuna_tokenizer.encode(formatted_prompt, return_tensors="pt")
     output_ids = vicuna_model.generate(input_ids, do_sample=True, max_length=149, num_beams=5, no_repeat_ngram_size=2)
@@ -70,7 +70,7 @@ def vicuna_respond(tab_name, message, chat_history):
 
 def llama_respond(tab_name, message, chat_history):
     formatted_prompt = f'''Generate the output only for the assistant. Please output any {tab_name} in the following sentence one per line without any additional text: {message}'''
-    print('Prompt + Context:')
+    print('Llama - Prompt + Context:')
     print(formatted_prompt)
     input_ids = llama_tokenizer.encode(formatted_prompt, return_tensors="pt")
     output_ids = llama_model.generate(input_ids, do_sample=True, max_length=149, num_beams=5, no_repeat_ngram_size=2)