r0r-5chach
/
dermy-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adjusted when file deleted

This commit is contained in:
Joshua Perry 2024-06-09 18:20:45 +01:00
parent 0d8d48b1aa
commit 4392d15029
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -22,6 +22,7 @@ model_bytes = open("models/mobilenet_v3.keras", "rb")
# Convert model to Keras model # Convert model to Keras model
model_file = open("model.keras", "wb").write(model_bytes.read()) model_file = open("model.keras", "wb").write(model_bytes.read())
model = tf.keras.models.load_model("model.keras") model = tf.keras.models.load_model("model.keras")
os.remove("model.keras")
benign_image = preprocess_image_bytes(benign_image_bytes) benign_image = preprocess_image_bytes(benign_image_bytes)
@ -30,7 +31,6 @@ malignant_image = preprocess_image_bytes(malignant_image_bytes)
benign_prediction = model.predict(benign_image) benign_prediction = model.predict(benign_image)
malignant_prediction = model.predict(malignant_image) malignant_prediction = model.predict(malignant_image)
os.remove("model.keras")
print(f"Benign Prediction: {benign_prediction}") print(f"Benign Prediction: {benign_prediction}")
print(f"Malignant Prediction: {malignant_prediction}") print(f"Malignant Prediction: {malignant_prediction}")