jalFaizy commited on
Commit
3b5fa06
·
1 Parent(s): 8aa1703

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ def infer(filename):
99
  return result, plot_slices(2, 10, 128, 128, vol[0, :, :, :20])
100
 
101
  if __name__ == "__main__":
102
- model = from_pretrained_keras('jalFaizy/3D_CNN')
103
 
104
  inputs = gr.inputs.File()
105
  outputs = [gr.outputs.Textbox(), 'plot']
 
99
  return result, plot_slices(2, 10, 128, 128, vol[0, :, :, :20])
100
 
101
  if __name__ == "__main__":
102
+ model = from_pretrained_keras('keras-io/3D_CNN_Pneumonia')
103
 
104
  inputs = gr.inputs.File()
105
  outputs = [gr.outputs.Textbox(), 'plot']