RoAr777 commited on
Commit
0f685dd
1 Parent(s): c5a733e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from fastai.vision.all import *
4
  import matplotlib.pyplot as plt
5
 
6
  # Load the trained model
7
- learn = load_learner('emotion_model')
8
 
9
  # Define function to preprocess image
10
  def preprocess_image(image):
 
4
  import matplotlib.pyplot as plt
5
 
6
  # Load the trained model
7
+ learn = load_learner('emotion_model.pth')
8
 
9
  # Define function to preprocess image
10
  def preprocess_image(image):