Update app.py

This commit is contained in:
Saifeddine ALOUI 2024-03-18 10:38:17 +01:00 committed by GitHub
parent f912b69848
commit 9399e4fdd9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -197,7 +197,10 @@ class LollmsApplication(LoLLMsCom):
if self.config.debug:
ASCIIColors.yellow(prompt)
summarized_chunk = self.model.generate(prompt, max_tokens)
summarized_chunks.append(summarized_chunk.strip())
if summarized_chunk:
summarized_chunks.append(summarized_chunk.strip())
else:
raise Exception("Couldn't generate text.")
summarized_content = "\n".join(summarized_chunks)
return summarized_content