Hugging Face's logo

Spaces:
hrnph
/
Pork_doneness_classification
Runtime error

App Files Community
SHXn3 commited on
Commit
781db03
·
verified ·
1 Parent(s): 8258396

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from PIL import Image
10
  import torch.nn as nn
11
 
12
  # Paths
13
- BASE_DIR = os.path.dirname(os.path.abspath(__file__))
14
  MODEL_PATH = os.path.join(BASE_DIR, "src", "mobilenetv2.pth")
15
  CLASSES_PATH = os.path.join(BASE_DIR, "src", "classes.txt")
16
 
 
10
  import torch.nn as nn
11
 
12
  # Paths
13
+ BASE_DIR = os.getcwd()
14
  MODEL_PATH = os.path.join(BASE_DIR, "src", "mobilenetv2.pth")
15
  CLASSES_PATH = os.path.join(BASE_DIR, "src", "classes.txt")
16