Update app.py

app.py

@@ -38,7 +38,7 @@ Assistant: "girlfriend;mother;father;friend"
 
 @spaces.GPU
 def get_past_key_values(system_prompt):
-    messages = [{'role': 'system', 'content': system}]
+    messages = [{'role': 'system', 'content': system_prompt}]
     tokenized = tokenizer.apply_chat_template(messages, return_tensors='pt')
     
     # assert that this is indeed a prefix (TODO: make sure this is robust)