Spaces:
Sleeping
Sleeping
File size: 715 Bytes
5b2de46 def5b54 5b2de46 e131f6a 5b2de46 e131f6a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from django.shortcuts import render
from django.http import HttpResponse
from django.views.decorators.csrf import csrf_exempt
import json
from ModelLoader import Colgen1
model_dir="models/colgen1/v2/"
model=Colgen1(model_dir)
# Create your views here.
@csrf_exempt
def generate(request):
if request.method=='POST':
received_data = json.loads(request.body.decode("utf-8"))
print(received_data)
out=model.colorToHex(model.predict(received_data['color_names']))
print(out)
data={
"color_names":received_data["color_names"],
"colors":out
}
return HttpResponse(json.dumps(data))
return HttpResponse("Working")
|