Spaces:
Sleeping
Sleeping
amazinghaha
commited on
Update app.py
Browse files
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(
|
187 |
-
ALL_message2 = load_from_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 =
|
194 |
LC_model = load_from_pkl(LC_model_Paht)['model'][0]
|
195 |
|
196 |
-
TF_model_Paht =
|
197 |
TF_model = load_from_pkl(TF_model_Paht)['model']
|
198 |
DR_model = load_from_pkl(TF_model_Paht)['dr']
|
199 |
|
200 |
-
Model_Paht =
|
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)
|