layai commited on
Commit
4bd94c4
·
1 Parent(s): e1a0355

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -0
app.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from huggingface_hub import from_pretrained_keras
3
+
4
+ def predict_diagnosis(loops):
5
+ loops_arr = tf.keras.utils.img_to_array(loops)
6
+ #new_loops_arr = np.array(loops)
7
+ new_img = tf.expand_dims(loops_arr,0)
8
+ model = from_pretrained_keras("layai/priya-loop-model")
9
+ single_data = model.predict(new_img)
10
+ single_data = single_data[0]
11
+ return np.argmax(single_data)
12
+
13
+ input = gr.inputs.Image(shape=(906, 255))
14
+ output = gr.outputs.Textbox(label='Predicted diagnosis')
15
+ iloops = gr.Interface( fn=predict_diagnosis,
16
+ inputs=input,
17
+ outputs=output,
18
+ capture_session=True)
19
+
20
+ iloops.launch()