MohamedRashad commited on
Commit
81d9f67
·
1 Parent(s): 0467139

Add image generation prompt and completion message to app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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': {