yiyii commited on
Commit
fc67e47
1 Parent(s): c207bc8
Files changed (1) hide show
  1. app.py +2 -1
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"Feel free to develop a complete story in depth and the generated story should approximately be {max_new_tokens} words long.[/INS]"
 
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)