Spaces:
Runtime error
Runtime error
File size: 1,282 Bytes
f6d3f4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
from io import BytesIO
import numpy as np
import tensorflow as tf
from PIL import Image
#from tensorflow.keras.applications.imagenet_utils import decode_predictions
from tensorflow.keras.models import load_model
import os
model = None
def load_model2():
model = load_model('application/models/resnet_ct.h5')
#model = load_model(os.path.join(modelpath, 'resnet_ct.h5'))
print("Model loaded")
return model
def predict(image: Image.Image):
global model
if model is None:
model = load_model2()
image = np.asarray(image.resize((224, 224)))[..., :3]
image = np.expand_dims(image, 0)
image = image / 127.5 - 1.0
result = model.predict(image)
probability = result[0]
#print("Resnet Predictions:")
if probability[0] > 0.5:
resnet_chest_pred = str('%.2f' % (probability[0]*100) + '% COVID')
else:
resnet_chest_pred = str('%.2f' % ((1-probability[0])*100) + '% NonCOVID')
response = []
for i, res in enumerate(result):
resp = {}
resp["prediction"] = resnet_chest_pred
#resp["confidence"] = f"{res[2]*100:0.2f} %"
response.append(resp)
return response
def read_imagefile(file) -> Image.Image:
image = Image.open(BytesIO(file))
return image
|