amazinghaha commited on
Commit
bbae99f
·
verified ·
1 Parent(s): 9f24646

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -183,21 +183,21 @@ class my_model7(nn.Module):
183
  Image_3D = None
184
  Current_name = None
185
 
186
- ALL_message = load_from_pkl(r'.\label0601.pkl')
187
- ALL_message2 = load_from_pkl(r'.\all_data_label.pkl')
188
  a = ALL_message2['train']
189
  a.update(ALL_message2['val'])
190
  a.update(ALL_message2['test'])
191
  ALL_message2 = a
192
 
193
- LC_model_Paht = r'.\train_ADA_1.pkl'
194
  LC_model = load_from_pkl(LC_model_Paht)['model'][0]
195
 
196
- TF_model_Paht = r'.\tf_model.pkl'
197
  TF_model = load_from_pkl(TF_model_Paht)['model']
198
  DR_model = load_from_pkl(TF_model_Paht)['dr']
199
 
200
- Model_Paht = r'./model_epoch120.pth.tar'
201
  checkpoint = torch.load(Model_Paht, map_location='cpu')
202
 
203
  classnet = my_model7(pretrained=False,num_classes=3,in_chans=1, img_size=224)
 
183
  Image_3D = None
184
  Current_name = None
185
 
186
+ ALL_message = load_from_pkl('./label0601.pkl')
187
+ ALL_message2 = load_from_pkl('./all_data_label.pkl')
188
  a = ALL_message2['train']
189
  a.update(ALL_message2['val'])
190
  a.update(ALL_message2['test'])
191
  ALL_message2 = a
192
 
193
+ LC_model_Paht = './train_ADA_1.pkl'
194
  LC_model = load_from_pkl(LC_model_Paht)['model'][0]
195
 
196
+ TF_model_Paht = './tf_model.pkl'
197
  TF_model = load_from_pkl(TF_model_Paht)['model']
198
  DR_model = load_from_pkl(TF_model_Paht)['dr']
199
 
200
+ Model_Paht = './model_epoch120.pth.tar'
201
  checkpoint = torch.load(Model_Paht, map_location='cpu')
202
 
203
  classnet = my_model7(pretrained=False,num_classes=3,in_chans=1, img_size=224)