Update server.py

server.py

@@ -51,7 +51,7 @@ def generate(prompt):
 def sample(prompt):
     # Sampling
     images = (
-        model.sampling(prompt=prompt, top_k=256, top_p=None, softmax_temperature=1.0, num_candidates=9, device=device)
+        model.sampling(prompt=prompt, top_k=96, top_p=None, softmax_temperature=1.0, num_candidates=9, device=device)
         .cpu()
         .numpy()
     )