Upload model

modeling.py

@@ -328,7 +328,7 @@ class CTCropModel(PreTrainedModel):
         x = self.normalize(x)
         # avg pooling
         features = F.adaptive_avg_pool2d(self.backbone(x), 1).flatten(1)
-        coords = self.linear(features).sigmoid()
+        coords = self.linear(self.dropout(features)).sigmoid()
 
         if add_buffer is not None:
             coords = self.add_buffer_to_coords(coords, add_buffer)