second
Browse files
app.py
CHANGED
@@ -145,7 +145,8 @@ def generate(image, temperature=0.9, max_new_tokens=1500, top_p=0.95, repetition
|
|
145 |
f"age: {age}, and gender: {gender} shown in the image. Begin with the scene described in the image's caption: '{image_caption}'. "
|
146 |
f"Just use the following example story plots and formats as an inspiration: "
|
147 |
f"{context} "
|
148 |
-
f"
|
|
|
149 |
)
|
150 |
|
151 |
temperature = float(temperature)
|
|
|
145 |
f"age: {age}, and gender: {gender} shown in the image. Begin with the scene described in the image's caption: '{image_caption}'. "
|
146 |
f"Just use the following example story plots and formats as an inspiration: "
|
147 |
f"{context} "
|
148 |
+
f"The generated story should include a beginning, middle, and end, and the complete story should approximately be {max_new_tokens} words.[/INS]"
|
149 |
+
# f"Feel free to develop a complete story in depth and the generated story should approximately be {max_new_tokens} words.[/INS]"
|
150 |
)
|
151 |
|
152 |
temperature = float(temperature)
|