This is a fine-tuned Machine Learning model trained to predict the outcome of appeal cases in the Indian Supreme Court. It takes the text of a court judgment as input and predicts whether the appeal will be Accepted or Rejected.
1import torch
2# If using a specific architecture class, import it first
3# from model_architecture import MyCustomModel
4
5# Path to the downloaded model file
6model_path = "entire_model.pt"
7
8# Load the model
9# Option A: If you saved the entire model
10model = torch.load(model_path)
11
12# Option B: If you saved only state_dict (Recommended)
13# model = MyCustomModel()
14# model.load_state_dict(torch.load(model_path))
15
16model.eval()
17print("Model loaded successfully!")