jerryjia92 commited on
Commit
73f728d
1 Parent(s): 4e30abb

alter app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,13 +1,13 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
 
 
 
4
  learn = load_learner('export.pkl')
5
 
6
  labels = learn.dls.vocab
7
 
8
- def is_cat(x):
9
- return 'cat' if (x.name)[0].isupper() else 'dog'
10
-
11
  def predict(img):
12
  img = PILImage.create(img)
13
  pred,pred_idx,probs = learn.predict(img)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
+ def is_cat(x):
5
+ return 'cat' if (x.name)[0].isupper() else 'dog'
6
+
7
  learn = load_learner('export.pkl')
8
 
9
  labels = learn.dls.vocab
10
 
 
 
 
11
  def predict(img):
12
  img = PILImage.create(img)
13
  pred,pred_idx,probs = learn.predict(img)