Update app.py

app.py

@@ -51,8 +51,8 @@ with gr.Blocks(
         <img src="https://github.com/viddexa/moderators/releases/download/v0.1.1/logo-v2.jpeg" alt="Visual Content Moderation" class="header-logo">
         <p>Official demo for the <a href="https://github.com/viddexa/moderators" target="_blank">moderators</a> package</p>
         <p>
-            <a href="https://huggingface.co/viddexa/nsfw-mini" target="_blank">Model: nsfw-mini</a> | 
-            <a href="https://huggingface.co/viddexa/nsfw-nano" target="_blank">Model: nsfw-nano</a> | 
+            <a href="https://huggingface.co/viddexa/nsfw-detection-2-mini" target="_blank">Model: nsfw-detection-2-mini</a> | 
+            <a href="https://huggingface.co/viddexa/nsfw-detection-2-nano" target="_blank">Model: nsfw-detection-2-nano</a> | 
             <a href="https://arxiv.org/abs/2312.16338" target="_blank">Arxiv</a> | 
             <a href="https://github.com/viddexa/moderators" target="_blank">GitHub</a> | 
             <a href="https://pypi.org/project/moderators/" target="_blank">PyPI</a>
@@ -78,8 +78,8 @@ with gr.Blocks(
     with gr.Row():
         with gr.Column(scale=1):
             model_choice = gr.Dropdown(
-                choices=["viddexa/nsfw-detection-mini", "viddexa/nsfw-detection-nano"],
-                value="viddexa/nsfw-detection-mini",
+                choices=["viddexa/nsfw-detection-2-mini", "viddexa/nsfw-detection-2-nano"],
+                value="viddexa/nsfw-detection-2-mini",
                 label="Model",
             )
             
@@ -92,7 +92,7 @@ with gr.Blocks(
             analyze_btn = gr.Button("Analyze", variant="primary")
         
         with gr.Column(scale=1):
-            output = gr.Label(label="Classification Scores", num_top_classes=10)
+            output = gr.Label(label="Classification Scores", num_top_classes=5)
     
     gr.Markdown("### 🎯 Try an Example")
     gr.Examples(