Update app.py

app.py

@@ -4,7 +4,7 @@ from text2tags import TaggerLlama
 model = TaggerLlama()
 
 
-def predict(caption, max_tokens=128, temperature=0.8, top_k=0.8, top_p=0.95, repeat_penalty=1.1):
+def predict(caption, max_tokens=128, temperature=0.8, top_k=40, top_p=0.95, repeat_penalty=1.1):
     tags = model.predict_tags(caption, max_tokens=max_tokens, temperature=temperature,
                               top_k=top_k, top_p=top_p, repeat_penalty=repeat_penalty)
     return ', '.join(tags)