Update app.py

app.py

@@ -39,6 +39,11 @@ def clean_chat_output(txt, prompt, interlocutor_names):
   output = output[:output.find(delimiter)]
   return output
 
+# GPT-J-6B API
+API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
+def query(payload):
+  response = requests.post(API_URL, json=payload)
+  return response.json()
 
 def chat(message, npc_name, initial_prompt, top_p, temperature, max_new_tokens, history=[]):
     interlocutor_names = ["Player", npc_name]