protloc-ai / encode_model.py
Tanoj22
Fix all import issues for deployment
1b09f56
raw
history blame contribute delete
219 Bytes
import base64
with open("models/best_residue_model.pt", "rb") as f:
data = base64.b64encode(f.read()).decode()
with open("models/model_b64.py", "w") as f:
f.write('MODEL_DATA = "' + data + '"')
print("Done!")