Update app.py

app.py

@@ -73,9 +73,9 @@ def process_image_str(groupImageFilePath: str):
     
     for image, location in zip(images, locations):
         # Process the image
-        intputTensor = transforms(image).unsqueeze(0) # unsqueeze? add batch dimension??
+        intputTensor = transforms(image).unsqueeze(0)
         
-        # do AI stuff here
+        # Do AI stuff here and format output
         with torch.no_grad():
             outputs_t = model(intputTensor)
             cert, pred_t = torch.max(torch.softmax(outputs_t, dim=1), dim=1)