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