MohamedRashad commited on
Commit
c07df32
·
1 Parent(s): b54b529

Update t2i_client initialization to include HF_TOKEN from environment variables

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from trellis.utils import render_utils, postprocessing_utils
17
  from gradio_client import Client
18
 
19
  llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
20
- t2i_client = Client("black-forest-labs/FLUX.1-dev")
21
 
22
  def generate_t2i_prompt(item_name):
23
  llm_prompt_template = """You are tasked with creating a concise yet highly detailed description of an item to be used for generating an image in a game development pipeline. The image should show the **entire item** with no parts cropped or hidden. The background should always be plain and monocolor, with no focus on it.
 
17
  from gradio_client import Client
18
 
19
  llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
20
+ t2i_client = Client("black-forest-labs/FLUX.1-dev", hf_token=os.getenv("HF_TOKEN"))
21
 
22
  def generate_t2i_prompt(item_name):
23
  llm_prompt_template = """You are tasked with creating a concise yet highly detailed description of an item to be used for generating an image in a game development pipeline. The image should show the **entire item** with no parts cropped or hidden. The background should always be plain and monocolor, with no focus on it.