bhushanp commited on
Commit
eec986c
·
1 Parent(s): 117fe52

file path errors

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -3,10 +3,9 @@ from fastai.vision.all import *
3
  import os
4
  # Load a pre-trained image classification model
5
  root = os.path.dirname(__file__)
6
- try:
7
- learn = load_learner(os.path.join(root, "models", "model.pth"))
8
- except FileNotFoundError:
9
- print(os.listdir())
10
 
11
  # Function to make predictions from an image
12
  def classify_image(image):
 
3
  import os
4
  # Load a pre-trained image classification model
5
  root = os.path.dirname(__file__)
6
+
7
+ learn = load_learner(os.path.join(root, "models", "model.pkl"))
8
+
 
9
 
10
  # Function to make predictions from an image
11
  def classify_image(image):