linoyts HF Staff commited on
Commit
2afde1e
·
verified ·
1 Parent(s): f6c099a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ from safetensors.torch import load_file
18
 
19
  import os
20
  import base64
 
21
  import json
22
 
23
  SYSTEM_PROMPT = '''
 
18
 
19
  import os
20
  import base64
21
+ import BytesIO
22
  import json
23
 
24
  SYSTEM_PROMPT = '''