Update modeling_cameroon_meals.py
Browse files
modeling_cameroon_meals.py
CHANGED
|
@@ -34,11 +34,6 @@ class CameroonMealsModel(PreTrainedModel):
|
|
| 34 |
|
| 35 |
model_path = hf_hub_download(repo_id="paulinusjua/cameroon-meals", filename="cameroon_meals_full_model.pth")
|
| 36 |
self.model = torch.load(model_path, map_location="cpu", weights_only=False)
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
self.model = torch.nn.Module()
|
| 40 |
-
self.model.forward = self.model.forward
|
| 41 |
-
self.model.load_state_dict(self.model.state_dict())
|
| 42 |
self.model.eval()
|
| 43 |
|
| 44 |
def forward(self, x):
|
|
|
|
| 34 |
|
| 35 |
model_path = hf_hub_download(repo_id="paulinusjua/cameroon-meals", filename="cameroon_meals_full_model.pth")
|
| 36 |
self.model = torch.load(model_path, map_location="cpu", weights_only=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
self.model.eval()
|
| 38 |
|
| 39 |
def forward(self, x):
|