Alex Gonzalez commited on
Commit
af3e0f6
·
1 Parent(s): ed986c3

Added Model and Prediction Code

Browse files
Files changed (2) hide show
  1. app.py +6 -1
  2. 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
- return "Prediction!"
 
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