Update app.py

app.py

@@ -18,7 +18,7 @@ app = FastAPI()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def load_model():
-    config = read_params(config_path)
+    # config = read_params(config_path)
     model = timm.create_model('convnext_base.clip_laiona', pretrained=True, num_classes=3)
     model_state_dict = torch.load('model.pth', map_location=device)
     model.load_state_dict(model_state_dict)