Update app.py

app.py

@@ -13,7 +13,7 @@ import threading
 import concurrent.futures
 
 # 加载类别名称
-with open('output/class_names.pkl', 'rb') as f:
+with open('class_names.pkl', 'rb') as f:
     class_names = pickle.load(f)
 
 # 初始化模型
@@ -23,7 +23,7 @@ model.fc = nn.Sequential(
     nn.Dropout(0.2),
     nn.Linear(model.fc.in_features, len(class_names))
 )
-model.load_state_dict(torch.load('output/best_model.pth', map_location=device))
+model.load_state_dict(torch.load('best_model.pth', map_location=device))
 model = model.to(device)
 model.eval()
 
@@ -137,12 +137,12 @@ def predict_realtime(video_frame, remove_bg):
 
 def create_interface():
     examples = [
-        "data/r0_0_100.jpg",
-        "data/r0_18_100.jpg",
-        "data/9_100.jpg",
-        "data/127_100.jpg",
-        "data/5ecc819f1a579f513e0a1500fabb3f0.png",
-        "data/1105.jpg"
+        "r0_0_100.jpg",
+        "r0_18_100.jpg",
+        "9_100.jpg",
+        "127_100.jpg",
+        "5ecc819f1a579f513e0a1500fabb3f0.png",
+        "1105.jpg"
     ]
 
     with gr.Blocks(title="Fruit Classification", theme=gr.themes.Soft()) as demo: