Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ def predict(img):
|
|
11 |
model = tf.keras.models.load_model(r"model.keras")
|
12 |
print("hi")
|
13 |
Retina_classes = ['DR', 'No_DR']
|
14 |
-
img = np.resize(img,(224,224))
|
15 |
img = np.expand_dims(img, axis=0)
|
16 |
prediction=model.predict(img)[0]
|
17 |
print(prediction)
|
|
|
11 |
model = tf.keras.models.load_model(r"model.keras")
|
12 |
print("hi")
|
13 |
Retina_classes = ['DR', 'No_DR']
|
14 |
+
img = np.resize(img,(224,224,3))
|
15 |
img = np.expand_dims(img, axis=0)
|
16 |
prediction=model.predict(img)[0]
|
17 |
print(prediction)
|