Phauglin commited on
Commit
f06adfd
·
verified ·
1 Parent(s): 6698ec3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def clear_outputs():
98
  }
99
 
100
  # Load the AI model
101
- learn = load_learner('export.pkl')
102
 
103
  # Create the web interface
104
  with gr.Blocks() as demo:
 
98
  }
99
 
100
  # Load the AI model
101
+ learn = load_learner('resnet50_30_categories.pkl')
102
 
103
  # Create the web interface
104
  with gr.Blocks() as demo: