hiikate commited on
Commit
98d22ec
1 Parent(s): ba20424

added title and description

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -3,9 +3,13 @@
3
 
4
  # %%
5
  #/export
 
6
  from fastai.vision.all import *
7
  import gradio as gr
8
 
 
 
 
9
  def is_cat(x): return x[0].isupper()
10
 
11
  # %%
@@ -14,7 +18,7 @@ learn = load_learner('model.pkl')
14
 
15
  # %%
16
  #/export
17
- categories = ('Cat','Dog')
18
 
19
  def classify_image(img):
20
  pred,idx,probs = learn.predict(img)
 
3
 
4
  # %%
5
  #/export
6
+ from turtle import title
7
  from fastai.vision.all import *
8
  import gradio as gr
9
 
10
+ title = "Image Classifier"
11
+ description = "An image classifier trained on the Oxford Pets dataset with fastai. Created as a demo for Gradio and HuggingFace Spaces"
12
+
13
  def is_cat(x): return x[0].isupper()
14
 
15
  # %%
 
18
 
19
  # %%
20
  #/export
21
+ categories = ('Dog','Cat')
22
 
23
  def classify_image(img):
24
  pred,idx,probs = learn.predict(img)