Spaces:
Sleeping
Sleeping
Update openAI_manager.py
Browse files- openAI_manager.py +2 -1
openAI_manager.py
CHANGED
@@ -78,12 +78,13 @@ def generateTestSentences(model_name, group_kwds, attribute_kwds, progress):
|
|
78 |
num_steps = len(group_kwds)*len(attribute_kwds)
|
79 |
for gi, grp_kwd in enumerate(group_kwds):
|
80 |
for ai, att_kwd in enumerate(attribute_kwds):
|
81 |
-
progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"Generating {grp_kwd}<>{att_kwd}...")
|
82 |
|
83 |
kwd_pair = [grp_kwd.strip(), att_kwd.strip()]
|
84 |
|
85 |
gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
|
86 |
print(f"Gens for pair: <{kwd_pair}> -> {gens}")
|
|
|
87 |
all_gens.extend(gens)
|
88 |
|
89 |
return all_gens
|
|
|
78 |
num_steps = len(group_kwds)*len(attribute_kwds)
|
79 |
for gi, grp_kwd in enumerate(group_kwds):
|
80 |
for ai, att_kwd in enumerate(attribute_kwds):
|
81 |
+
# progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"Generating {grp_kwd}<>{att_kwd}...")
|
82 |
|
83 |
kwd_pair = [grp_kwd.strip(), att_kwd.strip()]
|
84 |
|
85 |
gens = genChatGPT(model_name, kwd_pair, num2gen, numTries, fixed_shots, temperature=0.8)
|
86 |
print(f"Gens for pair: <{kwd_pair}> -> {gens}")
|
87 |
+
progress((gi*len(attribute_kwds)+ai)/num_steps, desc=f"{gens}")
|
88 |
all_gens.extend(gens)
|
89 |
|
90 |
return all_gens
|