Hugging Face's logo

Spaces:
OllieFox
/
cxr_classifier
Runtime error

App Files Community
foxojs commited on
Commit
7943b1b
·
1 Parent(s): 63d254a

"update"

Browse files
Files changed (2) hide show
  1. app.py +1 -2
  2. my_exported_model.pkl +3 -0
app.py CHANGED
@@ -5,8 +5,7 @@ path = "my_exported_model.pkl"
5
  learn = load_learner(path)
6
 
7
  def classify_image(image):
8
- processed_image = preprocess_image(image)
9
- pred, idx, probs = learn.predict(processed_image)
10
  return dict(zip(learn.dls.vocab, map(float, probs)))
11
 
12
  iface = gr.Interface(fn=classify_image, inputs="image", outputs=gr.Label(),
 
5
  learn = load_learner(path)
6
 
7
  def classify_image(image):
8
+ pred, idx, probs = learn.predict(image)
 
9
  return dict(zip(learn.dls.vocab, map(float, probs)))
10
 
11
  iface = gr.Interface(fn=classify_image, inputs="image", outputs=gr.Label(),
my_exported_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1546d02b3d8e8c9638e062299ec7f298dd24538faa78b6a67d5eb27db1a3131f
3
+ size 103083722