forked from erikbern/deep-fonts
-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
47 lines (36 loc) · 1.22 KB
/
server.py
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import os
import flask
import model
import numpy
import PIL, PIL.Image, PIL.PngImagePlugin
import StringIO
import jinja2
os.environ['THEANO_FLAGS'] = 'device=cpu'
import theano
model = model.Model(artificial_font=True)
model.try_load()
run_fn = model.get_run_fn()
app = flask.Flask('fonts')
env = jinja2.Environment(loader=jinja2.FileSystemLoader('static'))
@app.route('/')
def root():
template = env.get_template('index.html')
return template.render(d=model.d)
@app.route('/font.png')
def font():
input_i = numpy.array(map(float, flask.request.query_string.split(',')))
input_i = numpy.array([input_i for z in xrange(62)], dtype=theano.config.floatX)
input_j = numpy.array(range(62), dtype=numpy.int32)
pred = run_fn(input_i, input_j).reshape((62, 64, 64))
img = PIL.Image.new('L', (8 * 64, 8 * 64), 255)
for z in xrange(62):
x, y = z % 8, z // 8
img_char = PIL.Image.fromarray(numpy.uint8(((1.0 - pred[z]) * 255)))
img.paste(img_char, (x * 64, y * 64))
s = StringIO.StringIO()
img.save(s, format='PNG')
return flask.Response(response=s.getvalue(), mimetype='image/png')
def main():
app.run(host='0.0.0.0', debug=False)
if __name__ == '__main__':
main()