Spaces:
Runtime error
Runtime error
Update app.py
Browse filesBack to pretrained fast ai method approach
app.py
CHANGED
@@ -4,10 +4,10 @@ from fastai.learner import load_learner
|
|
4 |
import gradio as gr
|
5 |
import skimage
|
6 |
|
7 |
-
|
8 |
-
learn = load_learner(
|
9 |
-
hf_hub_download("
|
10 |
-
)
|
11 |
|
12 |
labels = learn.dls.vocab
|
13 |
def predict(img):
|
|
|
4 |
import gradio as gr
|
5 |
import skimage
|
6 |
|
7 |
+
learn = from_pretrained_fastai("kurianbenoy/course_v5_lesson2_pets_convnext_base_in22k")
|
8 |
+
#learn = load_learner(
|
9 |
+
# hf_hub_download("kurianbenoy/course_v5_lesson2_pets_convnext_base_in22k", "model.pkl")
|
10 |
+
#)
|
11 |
|
12 |
labels = learn.dls.vocab
|
13 |
def predict(img):
|