Update app.py

app.py

@@ -236,7 +236,7 @@ def count(image, text, prompts, state, device):
     input_image_exemplars = input_image_exemplars.unsqueeze(0).to(device)
     exemplars = [exemplars["exemplars"].to(device)]
 
-    print("model device: " + str(model.device))
+    print("model device: " + str(next(model.parameters()).device))
     print("input image device: " + str(input_image.device))
     print("input image exemplars device: " + str(input_image_exemplars.device))
     print("exemplars device: " + str(exemplars[0].device))