Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from fastai.vision.all import *
|
|
4 |
import gradio as gr
|
5 |
|
6 |
# Load the trained model
|
7 |
-
learn = load_learner('
|
8 |
|
9 |
# Define the categories based on your model's output
|
10 |
categories = learn.dls.vocab
|
|
|
4 |
import gradio as gr
|
5 |
|
6 |
# Load the trained model
|
7 |
+
learn = load_learner('second_model.pkl')
|
8 |
|
9 |
# Define the categories based on your model's output
|
10 |
categories = learn.dls.vocab
|