Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import torch.nn as nn
 import torchvision
 from model import create_model
+from pathlib import Path
 
 from timeit import default_timer as timer
 from typing import Tuple, Dict 
@@ -14,7 +15,7 @@ title = 'FoodVisionMini - Diljith'
 description = 'Classifies an image of food item into either one of 3 classes : pizza, steak or sushi'
 
 effnetb2, transforms, optimizer, lossFunc = create_model(num_classes = 3)
-effnetb2.load_state_dict(torch.load(f = 'effnet_b2-20%-10epochs.pth'), map_location = torch.device('cpu'))
+effnetb2.load_state_dict(torch.load(f = 'effnet_b2-20%-10epochs.pth', map_location = torch.device('cpu')))
 
 
 class_names = ['pizza', 'steak', 'sushi']
@@ -34,7 +35,7 @@ def predict(img):
     return probs_dict, end-start 
 
 
-
+examples_path = Path('examples/')
 example_list = [['examples/' + example] for example in os.listdir(examples_path)]
 
 projectApp = gr.Interface(