Update app.py

app.py

@@ -19,6 +19,7 @@ def get_adult_content(text):
   softmax = torch.nn.Softmax(dim=1)
   probs = softmax(logits)[0]
   probs = probs.cpu().detach().numpy()
+  
   max_index = np.argmax(probs)
   adult_content = labels[max_index]
   return adult_content