Update app.py

app.py

@@ -68,7 +68,7 @@ class ByT5ForTextGeotagging(PreTrainedModel):
         if return_coordinates:
             class_idx = torch.argmax(logits, dim=1).item()
             coordinates = self.config.class_to_location.get(str(class_idx))
-            confidence = torch.max(logits).item()
+            confidence = torch.max(torch.nn.functional.softmax(logits)).item()
             return logits, coordinates, confidence
         else:
             return logits