Update app.py

app.py

@@ -8,11 +8,16 @@ from PIL import Image
 # number convert to label
 labels = ["Drawings", "Hentai", "Neutral", "Porn", "Sexy"]
 description = """### NSFW 图片分类器 🚀  
-🤖 该模型可用于检测 NSFW（非工作安全）图片，并将其分类到 5 个类别。  
-📌 训练数据基于 [NSFW Data Scraper](https://github.com/alex000kim/nsfw_data_scraper) 处理。  
-🌟 代码仓库: [GitHub](https://github.com/csuer411/nsfw_classify)  
-📥 上传一张图片，AI 将预测其类别！"""
+该模型可以分类NSFW（非工作安全）图片，支持以下类别：  
+- **Drawings**：绘画图像  
+- **Hentai**：色情漫画  
+- **Neutral**：中立图像  
+- **Porn**：成人内容  
+- **Sexy**：性感内容  
 
+上传一张图片，我们将为您预测其类别。  
+更多信息请访问 [GitHub](https://github.com/csuer411/nsfw_classify)"""
+  
 # Define CNN model
 class Classifier(nn.Module):
     def __init__(self):
@@ -54,18 +59,57 @@ def predict(image_path):
     return result
 
 # 美化 Gradio 界面
-with gr.Blocks(theme="glass", css="footer {visibility: hidden}") as demo:
+css = """
+    .gradio-container {
+        font-family: 'Arial', sans-serif;
+        background-color: #f0f4f8;
+    }
+    .gr-button {
+        background-color: #4CAF50;
+        color: white;
+        font-weight: bold;
+    }
+    .gr-button:hover {
+        background-color: #45a049;
+    }
+    .gr-label {
+        font-size: 1.2em;
+        font-weight: bold;
+    }
+    .gr-image {
+        border: 2px solid #4CAF50;
+        border-radius: 8px;
+    }
+    .gr-row {
+        margin-bottom: 20px;
+    }
+    .gr-markdown {
+        color: #333;
+    }
+    .gr-input {
+        border-radius: 5px;
+    }
+    footer {visibility: hidden}
+"""
+
+with gr.Blocks(theme="huggingface", css=css) as demo:
     gr.Markdown("# 🌟 NSFW 图片分类器")
     gr.Markdown(description)
 
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(type="filepath", label="📷 上传图片")
+            image_input = gr.Image(type="filepath", label="📷 上传图片", elem_id="input-image")
             submit_btn = gr.Button("🚀 运行检测", variant="primary")
         
         with gr.Column():
             output_label = gr.Label(num_top_classes=2, label="📊 预测结果")
-
+            gr.Markdown("#### 示例图片")
+            examples = [
+                "./example/anime.jpg",
+                "./example/real.jpg"
+            ]
+            gr.Image(examples=examples, type="filepath", label="点击上传示例图片", elem_id="example-images")
+    
     submit_btn.click(predict, inputs=image_input, outputs=output_label)
 
 # 启动 Web 界面