Spaces:
Sleeping
Sleeping
okaris
commited on
Commit
•
51fbf43
1
Parent(s):
224d97a
HF Spaces
Browse files- requirements.txt +2 -0
- space.py +2 -0
requirements.txt
CHANGED
|
@@ -12,3 +12,5 @@ torchsde==0.2.6
|
|
| 12 |
transformers==4.37.1
|
| 13 |
onnxruntime-gpu
|
| 14 |
hf_transfer
|
|
|
|
|
|
|
|
|
| 12 |
transformers==4.37.1
|
| 13 |
onnxruntime-gpu
|
| 14 |
hf_transfer
|
| 15 |
+
gradio
|
| 16 |
+
spaces
|
space.py
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
import gradio as gr
|
|
|
|
| 2 |
from omni_zero import OmniZeroSingle
|
| 3 |
|
|
|
|
| 4 |
def generate(
|
| 5 |
seed=42,
|
| 6 |
prompt="A person",
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import spaces
|
| 3 |
from omni_zero import OmniZeroSingle
|
| 4 |
|
| 5 |
+
@spaces.GPU
|
| 6 |
def generate(
|
| 7 |
seed=42,
|
| 8 |
prompt="A person",
|