Update app.py

app.py

@@ -12,27 +12,27 @@ MAX_PROMPT_TOKENS = 30
 
 ## info
 model_info = {
-    'meta-llama/Llama-2-7b-chat-hf': dict(device_map='cpu', token=os.environ['hf_token'], 
+    'LLAMA2-7B': dict(model_path='meta-llama/Llama-2-7b-chat-hf', device_map='cpu', token=os.environ['hf_token'], 
                                           original_prompt_template='<s>[INST] {prompt} [/INST]',
                                           interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
                                          ), # , load_in_8bit=True
     
-    'google/gemma-2b': dict(device_map='cpu', token=os.environ['hf_token'],
+    'Gemma-2B': dict(model_path='google/gemma-2b', device_map='cpu', token=os.environ['hf_token'],
                             original_prompt_template='<bos> {prompt}',
                             interpretation_prompt_template='<bos>User: [X]\n\nAnswer: {prompt}',
                            ),
     
-    'mistralai/Mistral-7B-Instruct-v0.2': dict(device_map='cpu', 
+    'Mistral-7B Instruct': dict(model_path='mistralai/Mistral-7B-Instruct-v0.2', device_map='cpu', 
                                                original_prompt_template='<s>[INST] {prompt} [/INST]',
                                                interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
                                               ),
     
-    'TheBloke/Mistral-7B-Instruct-v0.2-GGUF': dict(model_file='mistral-7b-instruct-v0.2.Q5_K_S.gguf', 
-                                                   tokenizer='mistralai/Mistral-7B-Instruct-v0.2',
-                                                   model_type='llama', hf=True, ctransformers=True,
-                                                   original_prompt_template='<s>[INST] {prompt} [/INST]',
-                                                   interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
-                                                  )
+    # 'TheBloke/Mistral-7B-Instruct-v0.2-GGUF': dict(model_file='mistral-7b-instruct-v0.2.Q5_K_S.gguf', 
+    #                                                tokenizer='mistralai/Mistral-7B-Instruct-v0.2',
+    #                                                model_type='llama', hf=True, ctransformers=True,
+    #                                                original_prompt_template='<s>[INST] {prompt} [/INST]',
+    #                                                interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
+    #                                               )
         }