nelbarman053
commited on
Commit
•
aa39819
1
Parent(s):
c356d44
Update app.py
Browse files
app.py
CHANGED
@@ -23,8 +23,8 @@ def recognize_image(image_path):
|
|
23 |
label, _, probs = model.predict(image_path)
|
24 |
return dict(zip(labels, map(float, probs)))
|
25 |
|
26 |
-
inputs = gr.Image(
|
27 |
-
outputs = gr.Label()
|
28 |
|
29 |
examples = [
|
30 |
'test images/unknown_01.jpg',
|
@@ -48,7 +48,7 @@ examples = [
|
|
48 |
|
49 |
interface = gr.Interface(fn=recognize_image, inputs = inputs, outputs=outputs, examples = examples)
|
50 |
|
51 |
-
interface.launch(
|
52 |
|
53 |
|
54 |
|
|
|
23 |
label, _, probs = model.predict(image_path)
|
24 |
return dict(zip(labels, map(float, probs)))
|
25 |
|
26 |
+
inputs = gr.inputs.Image(shape=(224,224))
|
27 |
+
outputs = gr.outputs.Label()
|
28 |
|
29 |
examples = [
|
30 |
'test images/unknown_01.jpg',
|
|
|
48 |
|
49 |
interface = gr.Interface(fn=recognize_image, inputs = inputs, outputs=outputs, examples = examples)
|
50 |
|
51 |
+
interface.launch()
|
52 |
|
53 |
|
54 |
|