stmnk commited on
Commit
b35ca3b
·
1 Parent(s): 529af3f

fix? broken lnes

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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, \n f"duplicate_documents parameter must be {', '.join(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}":