File size: 667 Bytes
36d7307 f3618e8 e46e50a f3618e8 36d7307 f3618e8 820da0a f3618e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
import cv2
import numpy as np
from PIL import Image
from keras.preprocessing import image
import tensorflow as tf
#predict function
model = tf.keras.models.load_model(r"model.keras")
def predict(img):
print("hi")
print("hi")
Retina_classes = ['DR', 'No_DR']
img = np.resize(img,(224,224,3))
img = np.expand_dims(img, axis=0)
prediction=model.predict(img)[0]
print(prediction)
return {Retina_classes[i]: float(prediction[i]) for i in range(2)}
#Just four line implementation
image = gr.Image(label="Upload Here")
label = gr.Label(num_top_classes=2)
gr.Interface(fn=predict, inputs="image", outputs=label).launch() |