update model code

model.py

@@ -157,4 +157,4 @@ def check_file(image_path):
     model = nn.DataParallel(model)
     output = test_model(model, test_loader, device)
     print(output)
-    return str(output)
+    return "Control" if output.item() == 0 else "Axial"