sarth123 commited on
Commit
2f1f2e6
·
verified ·
1 Parent(s): 4fd5df8

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_adv.keras")
12
  print("hi")
13
  Retina_classes = ['DR', 'No_DR']
14
- img = np.resize(img,(299,299,3))
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_adv.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)