Update app.py

app.py

@@ -10,14 +10,15 @@ from tensorflow.keras.applications.inception_resnet_v2 import InceptionResNetV2
 #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
 model=load_model('best_model_76.h5')
 
+
 def classify_image(inp):
-  inp = inp.reshape((-1, IMG_SIZE, IMG_SIZE, 3))
-  #inp = tf.keras.applications.vgg16.preprocess_input(inp)
+  inp = inp.reshape((-1, HEIGHT,WIDTH, 3))
+  inp = tf.keras.applications.nasnet.preprocess_input(inp) 
   prediction = model.predict(inp).flatten()
   return {labels[i]: float(prediction[i]) for i in range(NUM_CLASSES)}
 
-image = gr.inputs.Image(shape=(IMG_SIZE, IMG_SIZE),label='Input')
-label = gr.outputs.Label(num_top_classes=2)
+image = gr.Image(shape=(HEIGHT,WIDTH),label='Input')
+label = gr.Label()
 
 gr.Interface(fn=classify_image, inputs=image, outputs=label, title='Brand Logo Detection').launch(debug=False)