Update app.py

app.py

@@ -13,7 +13,7 @@ MAX_PROMPT_TOKENS = 30
 ## info
 model_info = {
     'meta-llama/Llama-2-7b-chat-hf': dict(device_map='cpu', token=os.environ['hf_token'], 
-                                          original_prompt_template='<s>[INST] {prompt}',
+                                          original_prompt_template='<s>[INST] {prompt} [/INST]',
                                           interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
                                          ), # , load_in_8bit=True
     
@@ -111,6 +111,11 @@ css = '''
   margin-top: 10px;
   background-color: pink;
 }
+.bubble > textarea{
+  border: none;
+  background-color: inherit;
+}
+
 '''