Update app.py

app.py

@@ -96,7 +96,7 @@ testing_loader = DL(testSet, batch_size=VALID_BATCH_SIZE, shuffle=True)
 # To Strings for Use
 test_loader_strings = []
 for dat in testing_loader:
-     test_loader_strings += [d['input_ids'].tolist() for d in dat]
+     test_loader_strings += [d['input_ids'].tolist() for d in dat if isinstance(d, dict) and 'input_ids' in d]
 
 # model
 class DistilBERTClass(TNN.Module):