chansung commited on
Commit
65ccb66
·
1 Parent(s): b7908eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,6 +10,7 @@ import torch
10
  import gradio as gr
11
  from PIL import Image
12
  from datasets import load_dataset
 
13
  from diffusers import StableDiffusionPipeline
14
 
15
  from pingpong import PingPong
@@ -460,7 +461,7 @@ def share(title, editor, cover_art):
460
  if title.strip() == '':
461
  title = 'Untitled'
462
 
463
- dataset = load_dataset("chansung/llama2-stories")
464
 
465
  buffered = io.BytesIO()
466
  cover_art = Image.fromarray(cover_art.astype('uint8'))
 
10
  import gradio as gr
11
  from PIL import Image
12
  from datasets import load_dataset
13
+ from datasets import DownloadMode
14
  from diffusers import StableDiffusionPipeline
15
 
16
  from pingpong import PingPong
 
461
  if title.strip() == '':
462
  title = 'Untitled'
463
 
464
+ dataset = load_dataset("chansung/llama2-stories", download_mode=DownloadMode.FORCE_REDOWNLOAD)
465
 
466
  buffered = io.BytesIO()
467
  cover_art = Image.fromarray(cover_art.astype('uint8'))