chun commited on
Commit
0d1ab03
1 Parent(s): c792cc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
 
7
  def a(img):
8
  img = img.reshape( 64,64,3)
9
- model=load_model('./cats&dog.h5')
10
  test_image=np.expand_dims(img, axis=0)
11
  result=model.predict(test_image)
12
  if result[0][0]==1:
 
6
 
7
  def a(img):
8
  img = img.reshape( 64,64,3)
9
+ model=load_model('./demo.h5')
10
  test_image=np.expand_dims(img, axis=0)
11
  result=model.predict(test_image)
12
  if result[0][0]==1: