changed deprecated implementation

#1
by Y-T-G - opened
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -35,12 +35,8 @@ trained and tested on datasets procured from Cohen et al. and Stony Brook
35
  University. Just upload an image below and click "Submit" to obtain the
36
  model's prediction."""
37
  gr_interface = gr.Interface(fn=predict,
38
- inputs=gr.inputs.Image(
39
- shape=(224, 224)
40
- ),
41
- outputs=gr.outputs.Label(
42
- num_top_classes=len(labels)
43
- ),
44
  interpretation="default",
45
  title=title, description=description,
46
  examples=sample_xrays)
 
35
  University. Just upload an image below and click "Submit" to obtain the
36
  model's prediction."""
37
  gr_interface = gr.Interface(fn=predict,
38
+ inputs="image",
39
+ outputs="label",
 
 
 
 
40
  interpretation="default",
41
  title=title, description=description,
42
  examples=sample_xrays)