Update app.py

app.py

@@ -23,17 +23,17 @@ dataset_info = [
 
 model_info = {
     'LLAMA2-7B': dict(model_path='meta-llama/Llama-2-7b-chat-hf', device_map='cpu', token=os.environ['hf_token'], 
-                                          original_prompt_template='<s> {prompt}',
+                                          original_prompt_template='<s>{prompt}',
                                           interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
                                          ), # , load_in_8bit=True
     
     'Gemma-2B': dict(model_path='google/gemma-2b', device_map='cpu', token=os.environ['hf_token'],
-                            original_prompt_template='<bos> {prompt}',
+                            original_prompt_template='<bos>{prompt}',
                             interpretation_prompt_template='<bos>User: [X]\n\nAnswer: {prompt}',
                            ),
     
     'Mistral-7B Instruct': dict(model_path='mistralai/Mistral-7B-Instruct-v0.2', device_map='cpu', 
-                                               original_prompt_template='<s> {prompt}',
+                                               original_prompt_template='<s>{prompt}',
                                                interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
                                               ),
     
@@ -46,7 +46,7 @@ model_info = {
         }
 
 
-suggested_interpretation_prompts = ["Sure, I'll summarize your message:",
+suggested_interpretation_prompts = ["Sure, I'll summarize your message:", "Sure, here are the words in your message:",
                                     "Before responding, let me repeat the message you wrote:", 
                                     "Let me repeat the message:"]