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