Update app.py

app.py

@@ -42,16 +42,16 @@ def generate(
     top_k: int = 50,
     repetition_penalty: float = 1.2,
 ) -> Iterator[str]:
+    conversation = []
     conversation = [json.loads(os.getenv("PROMPT"))]
-    # conversation = []
-    for user, assistant in chat_history:
-        conversation.extend(
-            [
-                json.loads(os.getenv("PROMPT")),
-                {"role": "user", "content": user},
-                {"role": "assistant", "content": assistant},
-            ]
-        )
+    # for user, assistant in chat_history:
+    #     conversation.extend(
+    #         [
+    #             json.loads(os.getenv("PROMPT")),
+    #             {"role": "user", "content": user},
+    #             {"role": "assistant", "content": assistant},
+    #         ]
+    #     )
     conversation.append({"role": "user", "content": message})
 
     input_ids = tokenizer.apply_chat_template(conversation, add_generation_prompt=True, return_tensors="pt")