Update app.py

app.py

@@ -97,7 +97,7 @@ with gr.Blocks() as demo:
                 q = image_clean
                 q = q.reshape((64, 320, 3))
                 #print (q.min(), q.max())
-                yield i, q, submit_btn.update(visible=False)
+                yield i, q, gr.update(visible=False)
             yield i, q, submit_btn.update(visible=True)
             #with requests.post(url=API_ENDPOINT, data=data, timeout=600, stream=True) as r:
             #    i = 0
@@ -106,9 +106,9 @@ with gr.Blocks() as demo:
             #        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)
+            #        yield i, q, gr.update(visible=False)
+            #    yield i, q, gr.update(visible=True)
         except Exception as e:
-            yield 1000, 255*np.ones((64, 320, 3)), submit_btn.update(visible=True)
+            yield 1000, 255*np.ones((64, 320, 3)), gr.update(visible=True)
     submit_btn.click(fn=infer, inputs=inputs, outputs=outputs, concurrency_limit=1)
 demo.queue(max_size=20).launch()