Update app.py

edited fix for fiel path

app.py

@@ -6,6 +6,7 @@ import gradio as gr
 import os
 import torch
 
+from pathlib import Path
 from model import create_effnetb2_model
 from timeit import default_timer as timer
 from typing import Tuple, Dict
@@ -17,11 +18,11 @@ class_names = ['pizza', 'steak', 'sushi']
 effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes = 3)
 
 # Load save weights
+BASE_DIR = Path(__file__).resolve().parent
+WEIGHTS_PATH = BASE_DIR / "09_pretrained_effnetb2_pizza_steak_sush_20_percent.pth"
+
 effnetb2.load_state_dict(
-    torch.load(
-        f="/content/demos/foodvision_mini/09_pretrained_effnetb2_pizza_steak_sush_20_percent.pth",
-        map_location=torch.device("cpu") # load the model to the cpu
-    )
+    torch.load(WEIGHTS_PATH, map_location=torch.device("cpu"))
 )
 
 ### 3. Predict function ###