Update app.py

app.py

@@ -67,7 +67,7 @@ def classify_GPU(input_data):
 
     #==================================================================
 
-    number_of_batches = 100 # @param {type:"slider", min:1, max:100, step:1}
+    number_of_batches = 1 # @param {type:"slider", min:1, max:100, step:1}
     
     # @markdown NOTE: You can increase the number of batches on high-ram GPUs for better classification
     
@@ -90,6 +90,8 @@ def classify_GPU(input_data):
           out = model.generate(x,
                               1,
                               temperature=0.3,
+                              filter_logits_fn=top_k,
+                              filter_kwargs={'k': 1},
                               return_prime=False,
                               verbose=False)