Harsh-Jadhav commited on
Commit
708a9bc
1 Parent(s): 07e5a62

Update app.py

Browse files

Corrected model name

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from fastai.vision.all import *
3
 
4
  # Load the pre-trained model
5
- model_path = "assets/model.pkl" # Replace with the path to your model file
6
  learn = load_learner(model_path)
7
 
8
  # Define a function to make predictions on an image
 
2
  from fastai.vision.all import *
3
 
4
  # Load the pre-trained model
5
+ model_path = "assets/model-r34.pkl" # Replace with the path to your model file
6
  learn = load_learner(model_path)
7
 
8
  # Define a function to make predictions on an image