Update app.py

app.py

@@ -11,7 +11,7 @@ device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cp
 pil_to_tensor = transforms.ToTensor()
 
 # Define a dictionary to map image filenames to weight files
-image_to_weights = {'./weights/flolv2_UHDLL.pt','./weights/flolv2_all_111439.pt'}
+image_to_weights = ['./weights/flolv2_UHDLL.pt','./weights/flolv2_all_111439.pt']
 
 # Initial model setup (without weights)
 model = create_model()