Update app.py

app.py

@@ -10,7 +10,7 @@ from tensorflow.keras.layers import *
 
 
 devices =  tf.config.experimental.list_physical_devices('GPU')
-tf.config.experimental.set_virtual_device_configuration(devices[0],tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)]) 
+tf.config.experimental.set_virtual_device_configuration(devices[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)]) 
 
 labels= {'Subway': 0, 'Starbucks': 1,'McDonalds': 2,'Burger King': 3,'KFC': 4,'Other': 5}
 HEIGHT,WIDTH=224,224