Update utils.py

utils.py

@@ -160,7 +160,7 @@ def generate(prompt,crazy,k):
         newPrompt += " architecture"
 
     images = model.sampling(prompt=newPrompt,
-                            top_k=256,
+                            top_k=k,
                             top_p=None,
                             softmax_temperature=crazy,
                             num_candidates=num_candidates,
@@ -181,7 +181,7 @@ def generate(prompt,crazy,k):
     item = {}
     item['prompt'] = prompt
     item['crazy'] = crazy
-    item['k'] = 20
+    item['k'] = k
     item['image'] = Image.fromarray((result*255).astype(np.uint8))
     st.session_state.results.append(item)