added recall to chat

app.py

@@ -5,17 +5,18 @@ import requests
 API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
 headers = {"Authorization": "Bearer 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"}
 prompt = """
-me: I'm Twimbly Twombly. I leave in California.
+me: I'm Twimbly Twombly. I live in California. I'm a robot."""
+
+prev_chat = """
 you: Hello. How are you?
-me: I'm fine
-you: """
+me: I'm fine"""
 
 examples = [["how are you?"], ["hello"]]
 
 
 def chat_generate(word):
 #   print(f"**current reply")
-  p = prompt + word.lower() + "\n" + "me: "
+  p = prompt + prev_chat + "\n" + "you: " + word.lower() + "\n" + "me: "
   print(f"*****Inside chat_generate - Prompt is :{p}")
   json_ = {"inputs": p,
             "parameters":
@@ -28,17 +29,9 @@ def chat_generate(word):
   response = requests.post(API_URL, headers=headers, json=json_)
   output = response.json()
   output_tmp = output[0]['generated_text']
-  print(f"GPTJ response without splits is: {output_tmp}")
-  if "\n\n" not in output_tmp:
-    if output_tmp.find('.') != -1:
-      idx = output_tmp.find('.')
-      reply = output_tmp[:idx+1]
-    else:
-      idx = output_tmp.rfind('\n')
-      reply = output_tmp[:idx]
-  else:
-    reply = output_tmp.split("\n\n")[0] # +"."
+  reply = output_tmp.split("you:")[0] # +"."
   print(f"Chat Response being returned is: {reply}")
+  prev_chat = "you: " + word.lower() + "\n" + "me: " + reply
   return reply
 
 def text_to_image(reply):