Toptopito commited on
Commit
a848dc4
·
1 Parent(s): f365622

First version of app

app.py CHANGED
@@ -1,7 +1,21 @@
1
  import gradio as gr
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ from fastai.vision.all import *
 
4
 
5
+ # load model
6
+ learn = load_learner('model.pkl')
7
+
8
+ categories = ('CLEAR', 'PNEUMONIA')
9
+
10
+ def classify_image(img):
11
+ pred, idx, probs = learn.predict(img)
12
+ return dict(zip(categories, map(float, probs)))
13
+
14
+ examples = ['./unseen_data/NORMAL/NORMAL2-IM-1427-0001.jpeg',
15
+ './unseen_data/PNEUMONIA/person1946_bacteria_4874.jpeg',
16
+ './unseen_data/NORMAL/NORMAL2-IM-1430-0001.jpeg',
17
+ './unseen_data\PNEUMONIA\person1946_bacteria_4875.jpeg']
18
+
19
+ intf = gr.Interface(fn=classify_image, inputs=gr.Image(type='pil'), outputs=gr.Label(), examples=examples)
20
+
21
+ intf.launch(inline=False)
model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c83d4b20ba564f59d4ae44fee161a73d859c60645d703a6ce6f3d7318fc79ff2
3
+ size 87559486
pneumonia_prediction.ipynb ADDED
The diff for this file is too large to render. See raw diff
 
unseen_data/NORMAL/NORMAL2-IM-1427-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1430-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1431-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1436-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1437-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1438-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1440-0001.jpeg ADDED
unseen_data/NORMAL/NORMAL2-IM-1442-0001.jpeg ADDED
unseen_data/PNEUMONIA/person1946_bacteria_4874.jpeg ADDED
unseen_data/PNEUMONIA/person1946_bacteria_4875.jpeg ADDED
unseen_data/PNEUMONIA/person1947_bacteria_4876.jpeg ADDED
unseen_data/PNEUMONIA/person1949_bacteria_4880.jpeg ADDED
unseen_data/PNEUMONIA/person1950_bacteria_4881.jpeg ADDED
unseen_data/PNEUMONIA/person1951_bacteria_4882.jpeg ADDED
unseen_data/PNEUMONIA/person1952_bacteria_4883.jpeg ADDED
unseen_data/PNEUMONIA/person1954_bacteria_4886.jpeg ADDED