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'))
|