Spaces:
Runtime error
Runtime error
taskswithcode
commited on
Commit
·
3ce8ea6
1
Parent(s):
c2ca3f3
Fixes
Browse files- twc_openai_embeddings.py +3 -2
twc_openai_embeddings.py
CHANGED
|
@@ -14,6 +14,7 @@ class OpenAIModel:
|
|
| 14 |
def __init__(self):
|
| 15 |
self.debug = False
|
| 16 |
self.model_name = None
|
|
|
|
| 17 |
print("In OpenAI API constructor")
|
| 18 |
|
| 19 |
|
|
@@ -30,7 +31,7 @@ class OpenAIModel:
|
|
| 30 |
|
| 31 |
|
| 32 |
def compute_embeddings(self,input_file_name,input_data,is_file):
|
| 33 |
-
if (len(openai.api_key) == 0):
|
| 34 |
print("Open API key not set")
|
| 35 |
return [],[]
|
| 36 |
in_file = self.model_name + '.'.join(input_file_name.split('.')[:-1]) + "_embed.json"
|
|
@@ -59,7 +60,7 @@ class OpenAIModel:
|
|
| 59 |
return texts,embeddings
|
| 60 |
|
| 61 |
def output_results(self,output_file,texts,embeddings,main_index = 0):
|
| 62 |
-
if (len(openai.api_key) == 0):
|
| 63 |
print("Open API key not set")
|
| 64 |
return {}
|
| 65 |
# Calculate cosine similarities
|
|
|
|
| 14 |
def __init__(self):
|
| 15 |
self.debug = False
|
| 16 |
self.model_name = None
|
| 17 |
+
self.skip_key = True
|
| 18 |
print("In OpenAI API constructor")
|
| 19 |
|
| 20 |
|
|
|
|
| 31 |
|
| 32 |
|
| 33 |
def compute_embeddings(self,input_file_name,input_data,is_file):
|
| 34 |
+
if (len(openai.api_key) == 0 and not self.skip_key):
|
| 35 |
print("Open API key not set")
|
| 36 |
return [],[]
|
| 37 |
in_file = self.model_name + '.'.join(input_file_name.split('.')[:-1]) + "_embed.json"
|
|
|
|
| 60 |
return texts,embeddings
|
| 61 |
|
| 62 |
def output_results(self,output_file,texts,embeddings,main_index = 0):
|
| 63 |
+
if (len(openai.api_key) == 0 and not self.skip_key):
|
| 64 |
print("Open API key not set")
|
| 65 |
return {}
|
| 66 |
# Calculate cosine similarities
|