victor commited on
Commit
2ce38cd
·
1 Parent(s): 8957328

added lines of code

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -52,16 +52,23 @@ def classify_pokemon(img):
52
  return dict(zip(categories, map(float, probs)))
53
 
54
  # Gradio interface setup
 
 
55
  image = gr.Image(type='pil') # Image size should match your training data size (e.g., 128x128)
56
  label = gr.Label()
57
  examples = ['zapdos.jpg']
58
 
 
 
 
59
  # Set up Gradio interface
60
  intf = gr.Interface(
61
  fn=classify_pokemon,
62
  inputs=image,
63
  outputs=label,
64
- examples=examples
 
 
65
  )
66
 
67
  # Launch the app
 
52
  return dict(zip(categories, map(float, probs)))
53
 
54
  # Gradio interface setup
55
+ title = "Pokémon first gen classifier"
56
+ description = "Based on the famous scene (Who's that Pokémon) of the pokemon tv show, this neural network accurately classifies a pokemon image."
57
  image = gr.Image(type='pil') # Image size should match your training data size (e.g., 128x128)
58
  label = gr.Label()
59
  examples = ['zapdos.jpg']
60
 
61
+
62
+
63
+
64
  # Set up Gradio interface
65
  intf = gr.Interface(
66
  fn=classify_pokemon,
67
  inputs=image,
68
  outputs=label,
69
+ examples=examples,
70
+ title=title,
71
+ description=description
72
  )
73
 
74
  # Launch the app