Update app.py

app.py

@@ -27,6 +27,7 @@ def infer(
     num_completions = int(num_completions)
     temperature = float(temperature)
     top_p = float(top_p)
+    top_k = int(top_k)
     stop = stop.split(";")
     seed = seed
     
@@ -34,6 +35,7 @@ def infer(
     assert 1 <= num_completions <= 5
     assert 0.0 <= temperature <= 10.0
     assert 0.0 <= top_p <= 1.0
+    assert 1 <= top_k <= 1000
 
     if temperature == 0.0:
         temperature = 0.01