Update

app.py

@@ -62,7 +62,7 @@ def load_encoders():
 import numpy as np
 def b64(txt,vec):
     buffer = io.BytesIO()
-    np.savez_compressed(buffer, txt=txt, vec=vec)
+    torch.save((txt,vec), buffer)
     buffer.seek(0)
     encoded = base64.b64encode(buffer.getvalue()).decode('utf-8')
     return encoded
@@ -74,7 +74,7 @@ def convert(prompt):
         prompt = [prompt]
     txt = t5(prompt)
     vec = clip(prompt)
-    return b64(txt.cpu().numpy(),vec.cpu().numpy())
+    return b64(txt,vec)
 
 with gr.Blocks() as demo:
     gr.Markdown("""A workaround for flux-flax to fit into 40G VRAM""")