djairbee5 commited on
Commit
066233a
1 Parent(s): b73a2e9
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -71,10 +71,11 @@ model = load_model(model_path)
71
  # Create the Gradio interface
72
  iface = gr.Interface(
73
  fn=predict_image,
74
- inputs=gr.Image(height = 224, width= 224),
75
  outputs=gr.Label(num_top_classes=1),
76
  title="Image Classification",
77
- description="Upload an image to get the predicted label"
 
78
  )
79
  # Launch the interface
80
  iface.launch()
 
71
  # Create the Gradio interface
72
  iface = gr.Interface(
73
  fn=predict_image,
74
+ inputs=gr.Image(),
75
  outputs=gr.Label(num_top_classes=1),
76
  title="Image Classification",
77
+ description="Upload an image to get the predicted label",
78
+ allow_flagging="never",
79
  )
80
  # Launch the interface
81
  iface.launch()