Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
|
|
|
|
|
|
|
|
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'))
|