Update app.py

app.py

@@ -6,6 +6,11 @@ from tensorflow.keras.models import load_model
 import tensorflow_addons as tfa
 import os
 from tensorflow.keras.layers import *
+#import torch.cuda
+
+
+devices =  tf.config.experimental.list_physical_devices('GPU')
+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