Alex Gonzalez
commited on
Commit
·
af3e0f6
1
Parent(s):
ed986c3
Added Model and Prediction Code
Browse files- app.py +6 -1
- export.pkl +3 -0
app.py
CHANGED
@@ -1,8 +1,13 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
2 |
from PIL import Image
|
3 |
|
|
|
|
|
4 |
def predict(value: Image.Image) -> str:
|
5 |
-
|
|
|
6 |
|
7 |
with gr.Blocks() as demo:
|
8 |
with gr.Row():
|
|
|
1 |
import gradio as gr
|
2 |
+
from fastai.vision.all import *
|
3 |
+
|
4 |
from PIL import Image
|
5 |
|
6 |
+
learn_inf = load_learner("explort.pkl")
|
7 |
+
|
8 |
def predict(value: Image.Image) -> str:
|
9 |
+
pred,pred_idx,probs = learn_inf.predict(value)
|
10 |
+
return f"{pred} at {probs[pred_idx]}"
|
11 |
|
12 |
with gr.Blocks() as demo:
|
13 |
with gr.Row():
|
export.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:25f61071e6afeb4ab897cc6cef05cd47babf933580e7f1c99389c0e98155e3ee
|
3 |
+
size 47625470
|