Update app.py
Browse files
app.py
CHANGED
@@ -169,6 +169,8 @@ def generate(image, pdfs, temperature=0.9, max_new_tokens=1500, top_p=0.95, repe
|
|
169 |
# f"Feel free to develop a complete story in depth and the generated story should approximately be {max_new_tokens} words.[/INS]"
|
170 |
)
|
171 |
|
|
|
|
|
172 |
temperature = float(temperature)
|
173 |
if temperature < 1e-2:
|
174 |
temperature = 1e-2
|
|
|
169 |
# f"Feel free to develop a complete story in depth and the generated story should approximately be {max_new_tokens} words.[/INS]"
|
170 |
)
|
171 |
|
172 |
+
print("prompt:", prompt)
|
173 |
+
|
174 |
temperature = float(temperature)
|
175 |
if temperature < 1e-2:
|
176 |
temperature = 1e-2
|