taskswithcode commited on
Commit
07ce4ab
1 Parent(s): 6052767
Files changed (1) hide show
  1. twc_openai_embeddings.py +2 -1
twc_openai_embeddings.py CHANGED
@@ -19,7 +19,7 @@ class OpenAIModel:
19
 
20
 
21
  def init_model(self,model_name = None):
22
- print("OpanAI: Init model",model_name)
23
  try:
24
  openai.api_key = os.getenv("OPENAI_API_KEY")
25
  except:
@@ -33,6 +33,7 @@ class OpenAIModel:
33
  self.model_name = "text-similarity-ada-001"
34
  else:
35
  self.model_name = model_name
 
36
 
37
 
38
  def compute_embeddings(self,input_file_name,input_data,is_file):
 
19
 
20
 
21
  def init_model(self,model_name = None):
22
+ print("OpenAI: Init model",model_name)
23
  try:
24
  openai.api_key = os.getenv("OPENAI_API_KEY")
25
  except:
 
33
  self.model_name = "text-similarity-ada-001"
34
  else:
35
  self.model_name = model_name
36
+ print("OpenAI: Init model complete",model_name)
37
 
38
 
39
  def compute_embeddings(self,input_file_name,input_data,is_file):