Update app.py

app.py

@@ -6,12 +6,9 @@ from tensorflow.keras.models import load_model
 import tensorflow_addons as tfa
 import os
 from tensorflow.keras.layers import *
-#import torch.cuda
+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
 model=load_model('best_model.h5')