Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ pipe_t2i.to('cuda:0', torch.float16)
|
|
| 60 |
### gradio
|
| 61 |
rembg_session = rembg.new_session()
|
| 62 |
|
| 63 |
-
|
| 64 |
def generate(text, seed):
|
| 65 |
def fill_background(image):
|
| 66 |
image = np.array(image).astype(np.float32) / 255.0
|
|
@@ -91,7 +91,7 @@ def generate(text, seed):
|
|
| 91 |
image = fill_background(image)
|
| 92 |
return image
|
| 93 |
|
| 94 |
-
|
| 95 |
def render(image, mc_resolution=256, formats=["obj"]):
|
| 96 |
scene_codes = model(image, device=device)
|
| 97 |
mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
|
|
|
|
| 60 |
### gradio
|
| 61 |
rembg_session = rembg.new_session()
|
| 62 |
|
| 63 |
+
@spaces.GPU
|
| 64 |
def generate(text, seed):
|
| 65 |
def fill_background(image):
|
| 66 |
image = np.array(image).astype(np.float32) / 255.0
|
|
|
|
| 91 |
image = fill_background(image)
|
| 92 |
return image
|
| 93 |
|
| 94 |
+
@spaces.GPU
|
| 95 |
def render(image, mc_resolution=256, formats=["obj"]):
|
| 96 |
scene_codes = model(image, device=device)
|
| 97 |
mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
|