Update app.py

app.py

@@ -20,10 +20,10 @@ def get_adult_content(text):
   probs = softmax(logits)[0]
   probs = probs.cpu().detach().numpy()
   
-  max_index = np.argmax(probs)
-  adult_content = labels[max_index]
+  #max_index = np.argmax(probs)
+   
   return adult_content
-  
+  adult_content = f"{labels[0]} : {round(probs[0]*100,2)}  {labels[1]} : {round(probs[1]*100,2)}"
 demo = gr.Interface(get_adult_content, inputs = gr.inputs.Textbox(label= "Input your text here"),
                     outputs = gr.outputs.Textbox(label='Category'))