Update app.py

app.py

@@ -9,7 +9,6 @@ from huggingface_hub import hf_hub_download
 
 # DEFINE MODEL NAME
 model_name = "DinoVdeau-large-2024_04_03-with_data_aug_batch-size32_epochs150_freeze"
-checkpoint_name = "lombardata/" + model_name
 
 # IMPORT CLASSIFICATION MODEL
 def create_head(num_features , number_classes ,dropout_prob=0.5 ,activation_func =nn.ReLU):
@@ -28,12 +27,10 @@ class NewheadDinov2ForImageClassification(Dinov2ForImageClassification):
     def __init__(self, config: Dinov2Config) -> None:
         super().__init__(config)
 
-        self.num_labels = config.num_labels
-        self.dinov2 = Dinov2Model(config)
-
         # Classifier head
         self.classifier = create_head(config.hidden_size * 2, config.num_labels)
-
+        
+checkpoint_name = "lombardata/" + model_name
 model = NewheadDinov2ForImageClassification.from_pretrained(checkpoint_name)
 
 # IMPORT MODEL CONFIG PARAMETERS