edited

model.py

@@ -3,8 +3,12 @@ import gradio as gr
 from transformers import ConvNextForImageClassification, AutoImageProcessor
 from PIL import Image
 
+model = ConvNextForImageClassification.from_pretrained("facebook/convnext-base-224")
+
+# Redefine classifier for 23 classes
+model.classifier = torch.nn.Linear(in_features=1024, out_features=23)
+
 # Load model configuration and weights manually
-model = ConvNextForImageClassification.from_pretrained("facebook/convnext-base-224")  # Load the base model
 model.load_state_dict(torch.load("convnext_base_finetuned.pth", map_location="cpu"))  # Load your finetuned weights
 model.eval()
 

requirements.txt

@@ -1,3 +1,4 @@
 torch==1.13.1
 transformers==4.27.0
 gradio==3.0.1
+numpy<2