sarth123 commited on
Commit
820da0a
·
verified ·
1 Parent(s): 08726b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)