.

app.py

@@ -7,24 +7,31 @@ import os
 API_ENDPOINT = os.getenv('API_ENDPOINT')
 API_KEY = os.getenv('API_KEY')
 
-# setup
-gallery = gr.Gallery(label="Rendered Image", show_label=False, elem_id="gallery").style(grid=[1], height="auto")
+title = "<h1><center>Markup-to-Image Diffusion Models with Scheduled Sampling</center></h1>"
+description = "<center>Yuntian Deng, Noriyuki Kojima, Alexander M. Rush</center>"
 
-# infer
-def infer(latex): 
-    formula = latex
-    data = {'formula': formula, 'api_key': API_KEY}
-    with requests.post(url=API_ENDPOINT, data=data, timeout=600, stream=True) as r:
-        i = 0
-        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, 3))
-            i += 1
-            yield i, [q,]
-    
-title = "Markup-to-Image Diffusion Models with Scheduled Sampling"
-description="Yuntian Deng, Noriyuki Kojima, Alexander M. Rush"
-
-# launch
-gr.Interface(fn=infer, inputs=["text"], outputs=[gr.Slider(0, 1000, value=0, label='step (out of 1000)'), gallery],title=title,description=description).queue(concurrency_count=20, max_size=200).launch(enable_queue=True)
+with gr.Blocks() as demo:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Row():
+        with gr.Column(scale=2):
+            textbox = gr.Textbox(label="LaTeX", lines=1, max_lines=1, placeholder='Type LaTeX formula here and click "Generate"')
+            submit_btn = gr.Button("Generate", elem_id="btn")
+        with gr.Column(scale=3):
+            slider = gr.Slider(0, 1000, value=0, label='step (out of 1000)')
+            image = gr.Image(label="Rendered Image", show_label=False, elem_id="image")
+    inputs = [textbox]
+    outputs = [slider, image, submit_btn]
+    def infer(formula):
+        data = {'formula': formula, 'api_key': API_KEY}
+        with requests.post(url=API_ENDPOINT, data=data, timeout=600, stream=True) as r:
+            i = 0
+            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, 3))
+                i += 1
+                yield i, q, submit_btn.update(visible=False)
+            yield i, q, submit_btn.update(visible=True)
+    submit_btn.click(fn=infer, inputs=inputs, outputs=outputs)
+demo.queue(concurrency_count=20, max_size=200).launch(enable_queue=True)