Nina-HK commited on
Commit
78617ca
1 Parent(s): c6cc9b1
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -72,5 +72,13 @@ binary_interface_efn = gr.Interface(fn=classify_efn,
72
  examples=[['300104.png']]
73
  )
74
 
75
- interface_collection = gr.Interface.Collection([binary_interface_cnn, binary_interface_efn])
76
- interface_collection.launch()
 
 
 
 
 
 
 
 
 
72
  examples=[['300104.png']]
73
  )
74
 
75
+ demo = gr.Interface(fn=lambda inp: {"CNN": classify_cnn(inp), "EfficientNet": classify_efn(inp)},
76
+ inputs=gr.inputs.Image(shape=(224, 224)),
77
+ outputs=["json", "json"],
78
+ title="Binary Image Classification",
79
+ description="Classify an image as healthy or patient using either CNN or EfficientNet.",
80
+ examples=[['300104.png']]
81
+ )
82
+
83
+ if __name__ == "__main__":
84
+ demo.launch()