keerthi-balaji commited on
Commit
6967eff
·
1 Parent(s): ad2d334

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ def classify_image(image):
30
  # Create the Gradio interface
31
  iface = gr.Interface(
32
  fn=classify_image,
33
- inputs=gr.Image(shape=(224, 224)),
34
  outputs=gr.Label(num_top_classes=3),
35
  live=True,
36
  title="Image Classification",
@@ -38,4 +38,4 @@ iface = gr.Interface(
38
  )
39
 
40
  # Launch the Gradio interface
41
- iface.launch()
 
30
  # Create the Gradio interface
31
  iface = gr.Interface(
32
  fn=classify_image,
33
+ inputs=gr.Image(type="numpy"),
34
  outputs=gr.Label(num_top_classes=3),
35
  live=True,
36
  title="Image Classification",
 
38
  )
39
 
40
  # Launch the Gradio interface
41
+ iface.launch()