fix docker

model.py

@@ -22,12 +22,12 @@ class ChallengePromptGenerator:
         self.tokenizer = AutoTokenizer.from_pretrained(model_local_dir)
         
     def infer_prompt(
-        self, 
+        self,
         prompts,
         max_generation_length=77,
         beam_size=1,
         sampling_temperature=0.9,
-        sampling_topk=1,
+        sampling_topk=100,
         sampling_topp=1
     ):
         # Add bos
@@ -62,4 +62,4 @@ class ChallengePromptGenerator:
             if out[-1] != ".":
                 out = ".".join(out.split(".")[:-1]) + "."
             outputs.append(out)
-        return decoded_outputs
+        return outputs