Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
81d9f67
1
Parent(s):
0467139
Add image generation prompt and completion message to app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,6 @@ import os
|
|
9 |
os.environ['SPCONV_ALGO'] = 'native'
|
10 |
from typing import *
|
11 |
import torch
|
12 |
-
print(torch.__version__)
|
13 |
-
print(torch.cuda.is_available())
|
14 |
-
os.system("nvidia-smi")
|
15 |
import numpy as np
|
16 |
import imageio
|
17 |
import uuid
|
@@ -112,6 +109,9 @@ def generate_item_image(object_t2i_prompt):
|
|
112 |
trial_id, processed_image = preprocess_pil_image(image)
|
113 |
yield trial_id, processed_image
|
114 |
|
|
|
|
|
|
|
115 |
def pack_state(gs: Gaussian, mesh: MeshExtractResult, trial_id: str) -> dict:
|
116 |
return {
|
117 |
'gaussian': {
|
|
|
9 |
os.environ['SPCONV_ALGO'] = 'native'
|
10 |
from typing import *
|
11 |
import torch
|
|
|
|
|
|
|
12 |
import numpy as np
|
13 |
import imageio
|
14 |
import uuid
|
|
|
109 |
trial_id, processed_image = preprocess_pil_image(image)
|
110 |
yield trial_id, processed_image
|
111 |
|
112 |
+
generate_item_image("A vintage golden pocket watch with intricate floral engravings, polished metal, and Roman numerals on its clock face. Its chain is smooth and reflective, completing the elegant design.")
|
113 |
+
print("Done")
|
114 |
+
|
115 |
def pack_state(gs: Gaussian, mesh: MeshExtractResult, trial_id: str) -> dict:
|
116 |
return {
|
117 |
'gaussian': {
|