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

Update app.py

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