Estoy haciendo un proyecto de aprendizaje automático en el que necesito mostrar las predicciones en una página web. La página web está construida usando Django. Tengo la función de predicciones y los pesos del modelo, pero cómo integrar la función de predicciones, el modelo y los pesos en el código Django y hacer predicciones.

Mi código de predicción

 def predicting(model, device, loader): model.eval() total_preds = torch.Tensor() total_labels = torch.Tensor() with torch.no_grad(): for solute_graphs, solvent_graphs, solute_lens, solvent_lens in loader: outputs, i_map = model( [solute_graphs.to(device), solvent_graphs.to(device), torch.tensor(solute_lens).to(device), torch.tensor(solvent_lens).to(device)]) print(outputs) total_preds = torch.cat((total_preds, outputs.cpu()), 0) return total_preds.numpy().flatten()

Guardé los pesos en el archivo .tar , por lo que necesito ejecutar el modelo mientras cargaba los pesos para la predicción. No tengo idea de dónde guardar mi modelo PyTorch y los pesos para hacer inferencias usando Django. por favor ayuda.