Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def predict_jellyfish_type(uploaded_file):
|
|
15 |
with Image.open(uploaded_file) as img:
|
16 |
img = img.resize((150, 150)).convert('RGB') # Convert image to RGB
|
17 |
img_array = np.array(img)
|
18 |
-
|
19 |
|
20 |
prediction = model.predict(np.expand_dims(img_array, axis=0))
|
21 |
confidences = {labels[i]: np.round(float(prediction[0][i]), 2) for i in range(len(labels))}
|
|
|
15 |
with Image.open(uploaded_file) as img:
|
16 |
img = img.resize((150, 150)).convert('RGB') # Convert image to RGB
|
17 |
img_array = np.array(img)
|
18 |
+
|
19 |
|
20 |
prediction = model.predict(np.expand_dims(img_array, axis=0))
|
21 |
confidences = {labels[i]: np.round(float(prediction[0][i]), 2) for i in range(len(labels))}
|