Update app.py

app.py

@@ -82,7 +82,7 @@ class EmbeddingClassifier(nn.Module):
 class EmbeddingClassifierModel(nn.Module):
     def __init__(self, backbone_name, num_classes):
         super().__init__()
-        self.backbone = timm.create_model(backbone_name, pretrained=True, features_only=True)
+        self.backbone = timm.create_model(backbone_name, pretrained=True, num_classes=0,strict=False)
         self.data_config = timm.data.resolve_data_config({}, model=self.backbone)
         # This specific model uses a binary classifier with a single output logit
         # that indicates if an image is real (Non-AI). So, output_dim is fixed to 1.