Update app.py

app.py

@@ -93,6 +93,7 @@ with gr.Blocks() as demo:
                 #s = base64.b64encode(image_clean).decode('ascii')
                 #yield s
                 q = image_clean
+                q = q.reshape((64, 320, 3))
                 yield i, q, submit_btn.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: