Update app.py

app.py

@@ -18,9 +18,11 @@ def img2label(left, right):
     # 读取模型
     model = torch.load('densenet_FD_e4_l5e-4_b32.pkl', map_location='cpu').to(device)
 
+    # 模型推理
     with torch.no_grad():
         output = model(left=left_img.unsqueeze(0), right=right_img.unsqueeze(0))
 
+    # 结果处理
     output = torch.sigmoid(output.squeeze(0))
     output_ = output.cpu().numpy().tolist()
     res_dict = {LABELS[i]: output_[i] for i in range(len(output_))}
@@ -56,6 +58,7 @@ if __name__ == '__main__':
         transforms.Normalize(norm_mean, norm_std)
     ])
 
+    # 标签设置
     LABELS = {0: '正常',
               1: '糖尿病',
               2: '青光眼',