美化UI界面：添加新布局和样式

app.py

@@ -1,42 +1,33 @@
 import gradio as gr
 from ultralytics import YOLO
-import torch
-import cv2
-import numpy as np
-from fastapi import FastAPI, File, UploadFile
-from PIL import Image
-import io
-
-# 初始化 FastAPI
-app = FastAPI()
 
 # 加载模型
-model = YOLO("NailongKiller.yolo11n.pt")
+model = YOLO('NailongKiller.yolo11n.pt')
 
-def detect_objects(image):
-    if image is None:
-        return None
-    
-    try:
-        results = model(image)
-        result = results[0]
-        annotated_image = result.plot()
-        return cv2.cvtColor(annotated_image, cv2.COLOR_BGR2RGB)
-    except Exception as e:
-        return None
+def predict(img):
+    # 运行推理
+    results = model.predict(img)
+    # 返回第一个结果的绘制图像
+    return results[0].plot()
 
-# 创建 Gradio 界面
+# 创建 Gradio 演示
 demo = gr.Interface(
-    fn=detect_objects,
-    inputs=gr.Image(type="numpy"),
-    outputs=gr.Image(),
-    title="奶龙杀手 (Nailong Killer)",
-    description="上传图片来检测奶龙 | Upload an image to detect Nailong"
+    predict,
+    inputs=[
+        gr.Image(label="输入图片")
+    ],
+    outputs=[
+        gr.Image(label="检测结果", type="numpy")
+    ],
+    title="🐉 奶龙杀手 (NailongKiller)",
+    description="上传图片来检测奶龙 | Upload an image to detect Nailong",
+    examples=[
+        ["example1.jpg"],
+        ["example2.jpg"]
+    ],
+    cache_examples=True
 )
 
-# 将 Gradio 接口挂载到 FastAPI
-app = gr.mount_gradio_app(app, demo, path="/")
-
+# 启动应用
 if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    demo.launch()

requirements.txt

@@ -1,9 +1,5 @@
-ultralytics>=8.0.0
-torch>=2.0.0
-opencv-python-headless
-numpy>=1.22.2
-fastapi
-python-multipart
-uvicorn
-pillow
-gradio>=3.0.0
+gradio
+torch>=1.8.0
+torchvision>=0.9.0
+ultralytics>=8.3.0
+Pillow