CibeleGoudinho commited on
Commit
5ef9145
1 Parent(s): 55f53e3
Files changed (3) hide show
  1. app.py +2 -2
  2. rainy_sky.jpg +0 -0
  3. sunny_sky.jpg +0 -0
app.py CHANGED
@@ -6,7 +6,7 @@ warnings.simplefilter("ignore", category=UserWarning)
6
  learn = load_learner('model.pkl')
7
  warnings.resetwarnings()
8
 
9
- categories = ('Ensolarado', 'Chuvoso')
10
 
11
  def classify_image(img):
12
  _, _, probs = learn.predict(img)
@@ -15,7 +15,7 @@ def classify_image(img):
15
  image = gr.Image()
16
  label = gr.Label()
17
 
18
- examples = ['sunny_sky.jpg', 'rainy_sky.jpg']
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
21
  intf.launch(inline=False)
 
6
  learn = load_learner('model.pkl')
7
  warnings.resetwarnings()
8
 
9
+ categories = ('Orca', 'Estrela do mar')
10
 
11
  def classify_image(img):
12
  _, _, probs = learn.predict(img)
 
15
  image = gr.Image()
16
  label = gr.Label()
17
 
18
+ examples = ['k_whale.jpg', 'starfish.jpg']
19
 
20
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
21
  intf.launch(inline=False)
rainy_sky.jpg DELETED
Binary file (479 kB)
 
sunny_sky.jpg DELETED
Binary file (867 kB)