Update GPT4KG.py

GPT4KG.py

@@ -5,6 +5,7 @@ import numpy as np
 from sklearn.metrics.pairwise import cosine_similarity
 import torch
 import openai
+from openai import OpenAI
 from PIL import Image
 
 system_text = """You are an expert AI that extracts knowledge graphs from text and outputs JSON files with the extracted knowledge, and nothing more. Here's how the JSON is broken down.
@@ -110,7 +111,7 @@ class KnowledgeGraph:
       except:
         pass
       messages.append({"role":"user","content":text})
-      output = openai.ChatCompletion.create(model="gpt-4",messages=messages)["choices"][0]["message"].to_dict()["content"]
+      output = openai.ChatCompletion.create(model="gpt-4-1106-preview",messages=messages)["choices"][0]["message"].to_dict()["content"]
       return output
 
     def learn(self,text,show_output=False):