update model code

model.py

@@ -156,4 +156,5 @@ def check_file(image_path):
     model.to(device)
     model = nn.DataParallel(model)
     output = test_model(model, test_loader, device)
-    return output
+    print(output)
+    return str(output)