Spaces:
Sleeping
Sleeping
victor
commited on
Commit
·
2ce38cd
1
Parent(s):
8957328
added lines of code
Browse files
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
|