Spaces:
Runtime error
Runtime error
fix? broken lnes
Browse files
app.py
CHANGED
@@ -33,7 +33,8 @@ def write_documents(self, documents: Union[List[dict], List[Document]], index: O
|
|
33 |
if index is None:
|
34 |
index = self.index
|
35 |
duplicate_documents = duplicate_documents or self.duplicate_documents
|
36 |
-
assert duplicate_documents in self.duplicate_documents_options,
|
|
|
37 |
|
38 |
field_map = self._create_document_field_map()
|
39 |
document_objects = [Document.from_dict(d, field_map=field_map) if isinstance(d, dict) else d for d in documents]
|
@@ -184,9 +185,7 @@ def docgen_func(function_code, min_length, max_length, top_k, top_p, temp, repet
|
|
184 |
}
|
185 |
output = query(req_data)
|
186 |
if type(output) is list:
|
187 |
-
return f'"""
|
188 |
-
{output[0]["generated_text"]}
|
189 |
-
"""'
|
190 |
else:
|
191 |
msg = str(output)
|
192 |
if msg == "{'error': 'Model stmnk/codet5-small-code-summarization-python is currently loading', 'estimated_time': 20}":
|
|
|
33 |
if index is None:
|
34 |
index = self.index
|
35 |
duplicate_documents = duplicate_documents or self.duplicate_documents
|
36 |
+
assert duplicate_documents in self.duplicate_documents_options,
|
37 |
+
f"duplicate_documents parameter must be {', '.join(self.duplicate_documents_options)}"
|
38 |
|
39 |
field_map = self._create_document_field_map()
|
40 |
document_objects = [Document.from_dict(d, field_map=field_map) if isinstance(d, dict) else d for d in documents]
|
|
|
185 |
}
|
186 |
output = query(req_data)
|
187 |
if type(output) is list:
|
188 |
+
return f'"""{output[0]["generated_text"]}"""'
|
|
|
|
|
189 |
else:
|
190 |
msg = str(output)
|
191 |
if msg == "{'error': 'Model stmnk/codet5-small-code-summarization-python is currently loading', 'estimated_time': 20}":
|