Szeyu commited on
Commit
019caa7
·
verified ·
1 Parent(s): c98a781

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -40,11 +40,9 @@ def generate_content(image):
40
  raw = storyer(
41
  prompt,
42
  max_new_tokens=150,
43
- temperature=0.8,
44
- top_p=0.85,
45
- top_k=50,
46
- no_repeat_ngram_size=3,
47
- repetition_penalty=1.15,
48
  return_full_text=False
49
  )[0]["generated_text"].strip()
50
 
 
40
  raw = storyer(
41
  prompt,
42
  max_new_tokens=150,
43
+ temperature=0.7,
44
+ top_p=0.9,
45
+ no_repeat_ngram_size=2,
 
 
46
  return_full_text=False
47
  )[0]["generated_text"].strip()
48