MohamedRashad commited on
Commit
474e44a
·
1 Parent(s): 56ef76b

Remove unused imports and clean up generate_item_image function in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -18,7 +18,7 @@ from trellis.pipelines import TrellisImageTo3DPipeline
18
  from trellis.representations import Gaussian, MeshExtractResult
19
  from trellis.utils import render_utils, postprocessing_utils
20
  from gradio_client import Client
21
- from diffusers import FluxPipeline, AutoencoderKL, AutoencoderTiny
22
  from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
23
 
24
  llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
@@ -109,9 +109,6 @@ def generate_item_image(object_t2i_prompt):
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': {
 
18
  from trellis.representations import Gaussian, MeshExtractResult
19
  from trellis.utils import render_utils, postprocessing_utils
20
  from gradio_client import Client
21
+ from diffusers import FluxPipeline
22
  from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
23
 
24
  llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
 
109
  trial_id, processed_image = preprocess_pil_image(image)
110
  yield trial_id, processed_image
111
 
 
 
 
112
  def pack_state(gs: Gaussian, mesh: MeshExtractResult, trial_id: str) -> dict:
113
  return {
114
  'gaussian': {