debuging model loading

app.py

@@ -24,22 +24,23 @@ def load_class_labels(file_path='idx_to_class.json'):
         print(f"Error: {file_path} not found. Class labels will not be available.")
         return {}
 
-# Load model
-def load_model(model_path):
-    try:
-        model = timm.create_model('vit_base_patch16_224', pretrained=False, num_classes=47)
-        model.load_state_dict(torch.load(model_path, weights_only=True))
-        model.eval()
-        return model
-    except Exception as e:
-        print(f"Error loading model: {str(e)}")
-        return None
+
+
 
 # Global variables
 toxicity_data = load_toxicity_data()
 idx_to_class = load_class_labels()
 model_path = Path('vit_b16_224_25e_256bs_0.001lr_adamW_transforms.tar')
-model = load_model(model_path)
+model = None
+
+# Load model
+try:
+    model = timm.create_model('vit_base_patch16_224', pretrained=False, num_classes=47)
+    model.load_state_dict(torch.load(model_path, weights_only=True))
+    model.eval()
+except Exception as e:
+    print(f"Error loading model: {str(e)}")
+
 
 # Define the transformation
 transform = transforms.Compose([