Update app.py

app.py

@@ -127,25 +127,6 @@ AI:
 
 
 
-stop_tokens = [["Human", ":"], ["AI", ":"]]
-stopping_criteria = StoppingCriteriaList(
-    [StopGenerationCriteria(stop_tokens, tokenizer, model.device)]
-)
-
-
-
-class CleanupOutputParser(BaseOutputParser):
-    def parse(self, text: str) -> str:
-        user_pattern = r"\nUser"
-        text = re.sub(user_pattern, "", text)
-        human_pattern = r"\nHuman:"
-        text = re.sub(human_pattern, "", text)
-        ai_pattern = r"\nAI:"
-        return re.sub(ai_pattern, "", text).strip()
-
-    @property
-    def _type(self) -> str:
-        return "output_parser"
 
 
 template = """
@@ -156,19 +137,6 @@ Current conversation:
 
 Human: {input}
 AI:""".strip()
-prompt = PromptTemplate(input_variables=["history", "input"], template=template)
-
-memory = ConversationBufferWindowMemory(
-    memory_key="history", k=6, return_only_outputs=True
-)
-
-chain = ConversationChain(
-    llm=llm,
-    memory=memory,
-    prompt=prompt,
-    output_parser=CleanupOutputParser(),
-    verbose=True,
-)
 
 
 def get_llama_response(message: str, history: list) -> str: