from huggingface_hub import hf_hub_download
import json, torch
from lstm_model import LSTMClassifier
cfg = json.load(open(hf_hub_download("Lesia-Alba/nlp_reviews_lstm", "lstm_config.json")))
vocab = json.load(open(hf_hub_download("Lesia-Alba/nlp_reviews_lstm", "vocab_LSTM.json")))
weights = hf_hub_download("Lesia-Alba/nlp_reviews_lstm", "best_lstm_model.pt")
model = LSTMClassifier(
vocab_size=cfg["vocab_size"],
embedding_dim=cfg["embedding_dim"],
hidden_dim=cfg["hidden_dim"],
num_layers=cfg["num_layers"]
)
state = torch.load(weights, map_location="cpu")
model.load_state_dict(state)
model.eval()
