Skip to content

Instantly share code, notes, and snippets.

@att288
att288 / load_multiple_keras_models_django.py
Created May 4, 2019 17:24
load_multiple_keras_models_django.py
###################################################
# settings.py/the file where to load the models
###################################################
def load_model_from_path(path):
graph = tf.get_default_graph()
model = load_model(path)
return graph, model
def load_all_models():
global gModelObjs # each object is a tuple of graph, model
@att288
att288 / load_single_keras_django.py
Created May 4, 2019 17:08
load_single_keras_django.py
def _load_model_from_path(path):
global gGraph
global gModel
gGraph = tf.get_default_graph()
gModel = load_model(path) # keras function
_load_model_from_path('path_to_model')
@att288
att288 / load_keras_django_naive.py
Created May 4, 2019 17:06
load_keras_django_naive.py
import json
import os
import numpy as np
import pandas as pd
import tensorflow as tf
from keras import backend as K
from keras.models import load_model
from rest_framework.decorators import api_view