nelbarman053 commited on
Commit
aa39819
1 Parent(s): c356d44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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(height=512, width=512)
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(debug=True, share=True)
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