hf repo fixed

app.py

@@ -6,7 +6,7 @@ import os
 
 # Get model path (Windows compatible)
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-MODEL_ID = "dshi01/my-benthic-classifier"
+MODEL_ID = "dshi01/convnext-tiny-224-7clss"
 
 # Try different possible filenames
 # possible_names = ['ConvNextmodel.pth', 'convnextmodel.pth', 'ConvNext_model.pth']

testmodel.py

@@ -0,0 +1,4 @@
+from transformers import AutoModelForImageClassification, AutoImageProcessor
+m = AutoModelForImageClassification.from_pretrained("dshi01/convnext-tiny-224-7clss")
+p = AutoImageProcessor.from_pretrained("dshi01/convnext-tiny-224-7clss")
+print(m.config.model_type, m.config.num_labels)  # expect convnext, 7 (if set)