Update app.py

app.py

@@ -6,11 +6,11 @@ from PIL import Image
 
 def a(img):
     #img = img.reshape(1, 64, 64,3)
-    #img = img.reshape( 64, 64,3)
+    img = img.reshape( 64, 64,3)
     model=load_model('./cats&dog.h5')
     #test_image=image.load_img("pic01.jpg",target_size=(64,64))
-    test_image=image.img_to_array(img)
-    test_image=np.expand_dims(test_image, axis=0)    
+    #test_image=image.img_to_array(img)
+    test_image=np.expand_dims(img, axis=0)    
     result=model.predict(test_image)
     if result[0][0]==1:
         prediction='Dog'
@@ -29,12 +29,13 @@ inputs = [input]
 outputs = gr.outputs.Image(type="pil", label="Output Image")
 title = "Dog and Cat Object detection"
 
-demo=gr.Interface(a,
-             inputs, 
-             outputs, 
-             title=title,
-             examples=["a01.jpg", "a02.jpg","a03.jpg","a04.jpg"],
-             theme="huggingface").launch(enable_queue=True)
+
+
+image = gr.inputs.Image(shape=(64,64))
+
+
+demo=gr.Interface(fn=a, inputs=image,examples=["a01.jpg", "a02.jpg","a03.jpg","a04.jpg"],outputs="text").launch(debug='True')
+
 
 if __name__ == "__main__":
     demo.launch()