friesti1 commited on
Commit
0212f65
·
verified ·
1 Parent(s): d1e6dff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import tensorflow as tf
3
  from PIL import Image
4
  import numpy as np
5
 
6
- labels = ['Gengar', 'Seel', 'Zapdos']
7
 
8
  def predict_pokemon_type(uploaded_file):
9
 
@@ -28,7 +28,7 @@ iface = gr.Interface(
28
  inputs=gr.File(label="Upload File"), # File upload widget
29
  outputs="text", # Output type
30
  title="Pokemon Classifier", # Title of the interface
31
- description="Upload a picture of a pokemon (preferably Gengar, Seel, Zapdos)" # Description of the interface
32
  )
33
 
34
  # Launch the interface
 
3
  from PIL import Image
4
  import numpy as np
5
 
6
+ labels = ['Cubone', 'Ditto', 'Psyduck', 'Snorlax', 'Weedle']
7
 
8
  def predict_pokemon_type(uploaded_file):
9
 
 
28
  inputs=gr.File(label="Upload File"), # File upload widget
29
  outputs="text", # Output type
30
  title="Pokemon Classifier", # Title of the interface
31
+ description="Upload a picture of a pokemon (preferably Cubone, Ditto, Psyduck, Snorlax or Weedle)" # Description of the interface
32
  )
33
 
34
  # Launch the interface