concauu commited on
Commit
42a7cf2
·
verified ·
1 Parent(s): da01929

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -29,6 +29,7 @@ pipe = FluxPipeline.from_pretrained(
29
  torch_dtype=torch.bfloat16
30
  )
31
  pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 
32
 
33
  def enhance_prompt(user_prompt):
34
  """Enhances the given prompt using Groq and returns the refined prompt."""
 
29
  torch_dtype=torch.bfloat16
30
  )
31
  pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
32
+ groq_client = Groq(api_key="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")
33
 
34
  def enhance_prompt(user_prompt):
35
  """Enhances the given prompt using Groq and returns the refined prompt."""