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
|