Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
b7c04cb
1
Parent(s):
26b3ad9
zerogpu
Browse files- app.py +2 -0
- requirements.txt +4 -3
app.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
import gradio as gr
|
|
|
|
| 4 |
import torch
|
| 5 |
from diffusers import DiffusionPipeline
|
| 6 |
from gradio_client import Client, file
|
|
@@ -30,6 +31,7 @@ splat_pipeline = DiffusionPipeline.from_pretrained(
|
|
| 30 |
).to("cuda")
|
| 31 |
|
| 32 |
|
|
|
|
| 33 |
def run(input_image, convert):
|
| 34 |
input_image = input_image.astype("float32") / 255.0
|
| 35 |
images = image_pipeline(
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
+
import spaces
|
| 5 |
import torch
|
| 6 |
from diffusers import DiffusionPipeline
|
| 7 |
from gradio_client import Client, file
|
|
|
|
| 31 |
).to("cuda")
|
| 32 |
|
| 33 |
|
| 34 |
+
@spaces.GPU
|
| 35 |
def run(input_image, convert):
|
| 36 |
input_image = input_image.astype("float32") / 255.0
|
| 37 |
images = image_pipeline(
|
requirements.txt
CHANGED
|
@@ -19,6 +19,7 @@ safetensors
|
|
| 19 |
scikit-image
|
| 20 |
scikit-learn
|
| 21 |
scipy
|
|
|
|
| 22 |
tqdm
|
| 23 |
transformers
|
| 24 |
trimesh
|
|
@@ -26,7 +27,7 @@ kiui >= 0.2.3
|
|
| 26 |
xatlas
|
| 27 |
roma
|
| 28 |
plyfile
|
| 29 |
-
torch
|
| 30 |
-
torchvision
|
| 31 |
-
torchaudio
|
| 32 |
xformers
|
|
|
|
| 19 |
scikit-image
|
| 20 |
scikit-learn
|
| 21 |
scipy
|
| 22 |
+
spaces
|
| 23 |
tqdm
|
| 24 |
transformers
|
| 25 |
trimesh
|
|
|
|
| 27 |
xatlas
|
| 28 |
roma
|
| 29 |
plyfile
|
| 30 |
+
torch == 2.2.0
|
| 31 |
+
torchvision == 0.17.0
|
| 32 |
+
torchaudio == 2.2.0
|
| 33 |
xformers
|