Update app.py

app.py

@@ -35,7 +35,11 @@ def load_model():
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     
     # Model ve processor yükle (local files'dan)
-    model = ViTForImageClassification.from_pretrained("./", local_files_only=True)
+    model = ViTForImageClassification.from_pretrained(
+        "./", 
+        local_files_only=True,
+        num_labels=15,
+        ignore_mismatched_sizes=True)
     processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
     
     model.to(device)