.

app.py

@@ -19,7 +19,7 @@ def infer(latex):
         for line in r.iter_lines():
             response = line.decode('ascii').strip()
             r = base64.decodebytes(response.encode('ascii'))
-            q = np.frombuffer(r, dtype=np.float32).reshape((64, 320))
+            q = np.frombuffer(r, dtype=np.float32).reshape((64, 320, 3))
             i += 1
             yield i, [q,]