Update app.py

app.py

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