Spaces:

chansung
/

co-write-with-llama2

Runtime error

chansung commited on Aug 14, 2023

Commit

87bb15a

1 Parent(s): a4e09ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -463,8 +463,9 @@ def share(title, editor, cover_art):
   dataset = load_dataset("chansung/llama2-stories")
   buffered = io.BytesIO()
   cover_art.save(buffered, format="PNG")
-  img_str = base64.b64encode(buffered.getvalue())
   dataset['train'] = dataset['train'].add_item({
       'title': title,

   dataset = load_dataset("chansung/llama2-stories")
   buffered = io.BytesIO()
+  covert_art = Image.fromarray(cover_art.astype('uint8'))
   cover_art.save(buffered, format="PNG")
+  img_str = base64.b64encode(buffered.read())
   dataset['train'] = dataset['train'].add_item({
       'title': title,