ysharma HF staff commited on
Commit
5b304ce
1 Parent(s): 5a8dedb
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -62,11 +62,13 @@ def meme_generate(img): #prompt, generated_txt): #, input_prompt_sql ): #, input
62
  json_ = {"inputs": prompt,
63
  "parameters":
64
  {
65
- "top_p": 0.95,
66
- "top_K": 50,
67
  "max_new_tokens": 250,
 
68
  "num_return_sequences": 3,
69
  "return_full_text": True,
 
70
  },
71
  "options":
72
  {"use_cache": True,
 
62
  json_ = {"inputs": prompt,
63
  "parameters":
64
  {
65
+ #"top_p": 0.95,
66
+ "top_k": 50,
67
  "max_new_tokens": 250,
68
+ "temperature": 1.4,
69
  "num_return_sequences": 3,
70
  "return_full_text": True,
71
+ "do_sample": True,
72
  },
73
  "options":
74
  {"use_cache": True,