s1ri1337 commited on
Commit
6dd2514
1 Parent(s): 3e0f953

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -23,6 +23,7 @@ def run_model(img):
23
  #img_path = '/content/Indian/9/1020.jpg'
24
  #new_image = load_image(img_path)
25
  #result = model.predict(new_image)
 
26
  result = model.predict(img)
27
  results = dict(zip(classes, result[0]))
28
  return max(results, key = results.get)
 
23
  #img_path = '/content/Indian/9/1020.jpg'
24
  #new_image = load_image(img_path)
25
  #result = model.predict(new_image)
26
+ img = img.reshape((-1, 224, 224, 3))
27
  result = model.predict(img)
28
  results = dict(zip(classes, result[0]))
29
  return max(results, key = results.get)