Ashhar commited on
Commit
924b273
·
1 Parent(s): 73ee114

change the way flux client is created

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -295,8 +295,7 @@ def __generateImageHF(prompt: str):
295
 
296
 
297
  def __generateImage(prompt: str):
298
- fluxClient = replicate.Client(api_token=os.environ.get("REPLICATE_API_KEY"))
299
- result = fluxClient.run(
300
  "black-forest-labs/flux-schnell",
301
  input={
302
  "prompt": prompt,
 
295
 
296
 
297
  def __generateImage(prompt: str):
298
+ result = replicate.run(
 
299
  "black-forest-labs/flux-schnell",
300
  input={
301
  "prompt": prompt,