Fix prediction script
This commit is contained in:
parent
33c19f9112
commit
99aa9a85a0
@ -16,4 +16,4 @@ model = mlflow.pytorch.load_model("my_model")
|
|||||||
with open(f'my_model/{input}', 'r') as file:
|
with open(f'my_model/{input}', 'r') as file:
|
||||||
json_data = json.load(file)['inputs']
|
json_data = json.load(file)['inputs']
|
||||||
|
|
||||||
print(model(torch.tensor(np.array(json_data['inputs'])).float()))
|
print(model(torch.tensor(np.array(json_data['inputs'])).float()))
|
||||||
|
Loading…
Reference in New Issue
Block a user