.

app.py

@@ -26,14 +26,17 @@ with gr.Blocks() as demo:
     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)
+        try:
+            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)
+        except Exception as e:
+            yield 1000, 255*np.ones((64, 320, 3)), 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)