Daniel-Sousa commited on
Commit
225e617
1 Parent(s): 52bf9a8
Files changed (3) hide show
  1. Kidney_stones_2.pkl +3 -0
  2. app.py +26 -0
  3. requirements.txt +8 -0
Kidney_stones_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2908531802830404b5f4be981173ce60d201c87a642a12603abeff720fc8098f
3
+ size 418601
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pandas as pd
3
+ import numpy as np
4
+ import pickle
5
+
6
+ with open('Kidney_stones_2.pkl', 'rb') as arquivo:
7
+ modelo = pickle.load(arquivo)
8
+
9
+ def predict(gravity, ph, osmo, cond, urea, calc):
10
+ input = pd.DataFrame([[gravity, ph, osmo, cond, urea, calc]],
11
+ columns=['gravity', 'ph', 'osmo', 'cond', 'urea', 'calc'])
12
+ pred = modelo.predict(input)
13
+
14
+ if pred[0] == 0:
15
+ return "No Kidney Stones"
16
+ return "Kidney Stones"
17
+
18
+ interface = gr.Interface(
19
+ fn=predict,
20
+ inputs=["number", "number", "number", "number", "number", "number"],
21
+ outputs="text",
22
+ title="Verificador de Pedras nos Rins",
23
+ description="Esse modelo é capaz de realizar uma análise com base nos dados da urina de uma pessoa e identificar se ela possui pedra nos rins",
24
+ )
25
+
26
+ interface.launch(share=True)
requirements.txt ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ fastai
2
+ gradio
3
+ pip
4
+ pandas
5
+ numpy
6
+ scikit-learn==1.2.2
7
+ pickle
8
+ torch