Update app.py

app.py

@@ -12,9 +12,9 @@ import json
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Using {device} device")
 
-model_path = './model_danger_pose.pth'
-num_classes = 10
-label_name_list = ['c0', 'c1', 'c2', 'c3', 'c4', 'c5', 'c6', 'c7', 'c8', 'c9']
+model_path = './resnet18.pth'
+num_classes = 4
+label_name_list =['glioma_tumor', 'meningioma_tumor', 'no_tumor', 'pituitary_tumor']
 
 # step1: 创建并加载模型
 resnet18 = models.resnet18(pretrained=False)
@@ -71,7 +71,7 @@ demo = gr.Interface(fn=predict,
              inputs=gr.Image(type="pil"),
 
              outputs="text",
-             examples=["./pic1.jpg"],
+             examples=["./image_glioma_tumor.jpg"],
              )
              
 # demo.launch(debug=True)