small fix

app.py

@@ -58,9 +58,9 @@ def predict(prompt, image_1, image_2):
     print(f"Starting prediction for prompt: {prompt}")
     start_time = time.time()
     probs = calc_probs(prompt, [image_1, image_2])
-    print(f"Prediction: {probs:.3f} ({time.time() - start_time:.2f} seconds, ) ")
+    print(f"Prediction: {probs} ({time.time() - start_time:.2f} seconds, ) ")
     if device == "cuda":
-        print(f"GPU mem used: {round(torch.cuda.max_memory_allocated(device) / 1024 / 1024 / 1024, 2)} GB")
+        print(f"GPU mem used: {round(torch.cuda.max_memory_allocated(device) / 1024 / 1024 / 1024, 2)}/{round(torch.cuda.get_device_properties(device).total_memory / 1024 / 1024 / 1024, 2)} GB")
     return str(round(probs[0], 3)), str(round(probs[1], 3))