Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from diffusers import DiffusionPipeline
|
|
3 |
from PIL import Image
|
4 |
import numpy as np
|
5 |
|
6 |
-
pipeline = DiffusionPipeline.from_pretrained("Qilex/
|
7 |
|
8 |
def generate_pets(num_to_generate):
|
9 |
images = pipeline(num_to_generate)["sample"]
|
@@ -35,6 +35,8 @@ description = """
|
|
35 |
This AI will 'dream' you up a virtual pet.
|
36 |
\nThis is a denoising diffusion model trained in 48 hours for a hackathon, so the images can be pretty wonky.
|
37 |
\nImages are 128x128px.
|
|
|
|
|
38 |
"""
|
39 |
article = '''Here's a gallery of some of the better pets:
|
40 |
<div style="display: flex; justify-content:space-evenly">
|
|
|
3 |
from PIL import Image
|
4 |
import numpy as np
|
5 |
|
6 |
+
pipeline = DiffusionPipeline.from_pretrained("Qilex/VirtualPet-Gen")
|
7 |
|
8 |
def generate_pets(num_to_generate):
|
9 |
images = pipeline(num_to_generate)["sample"]
|
|
|
35 |
This AI will 'dream' you up a virtual pet.
|
36 |
\nThis is a denoising diffusion model trained in 48 hours for a hackathon, so the images can be pretty wonky.
|
37 |
\nImages are 128x128px.
|
38 |
+
|
39 |
+
\nBecause gradio isn't giving me a GPU, it takes 10-15 minutes to generate an image. Quick inference can be run in the <a href="https://colab.research.google.com/drive/19QtPOHv6HCpexyCMGXowX4vyZlF4ZZYN?usp=sharing">colab notebook</a>
|
40 |
"""
|
41 |
article = '''Here's a gallery of some of the better pets:
|
42 |
<div style="display: flex; justify-content:space-evenly">
|