Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ def generate_image(text):
|
|
40 |
|
41 |
# Constants
|
42 |
MAX_MAX_NEW_TOKENS = 2048
|
43 |
-
DEFAULT_MAX_NEW_TOKENS =
|
44 |
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
45 |
|
46 |
LICENSE = """
|
@@ -147,7 +147,7 @@ def generate(
|
|
147 |
final_story_trimmed = remove_last_sentence(final_story)
|
148 |
|
149 |
image_url = generate_image(final_story_trimmed)
|
150 |
-
|
151 |
|
152 |
|
153 |
def remove_last_sentence(text):
|
|
|
40 |
|
41 |
# Constants
|
42 |
MAX_MAX_NEW_TOKENS = 2048
|
43 |
+
DEFAULT_MAX_NEW_TOKENS = 1024
|
44 |
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
45 |
|
46 |
LICENSE = """
|
|
|
147 |
final_story_trimmed = remove_last_sentence(final_story)
|
148 |
|
149 |
image_url = generate_image(final_story_trimmed)
|
150 |
+
yield f"{final_story}\n\n"
|
151 |
|
152 |
|
153 |
def remove_last_sentence(text):
|