update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 
 from model import create_effnetb2_model
 from timeit import default_timer as timer
-from typing import Tuple, TypedDict
+from typing import Tuple, Dict
 
 # class_names
 class_names = ['pizza', 'steak', 'sushi']
@@ -19,7 +19,8 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
 # load saved weights
 effnetb2.load_state_dict(torch.load(
     f = '09_pretrained_effnet_feature_extractor_20_percent.pth',
-    map_location=torch.device('cpu')
+    map_location=torch.device('cpu'),
+    weights_only=True
     )
 )