Y-T-G commited on
Commit
5f748e4
1 Parent(s): 26c69f0

changed deprecated implementation

Browse files
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)