neo7team commited on
Commit
b98641e
·
verified ·
1 Parent(s): 73f54ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -13,7 +13,7 @@ import json
13
  API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
14
  API_TOKEN = os.getenv("HF_READ_TOKEN")
15
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
16
- timeout = 100
17
 
18
  # Function to query the API and return the generated image
19
  def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
@@ -25,13 +25,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
25
  API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN")])
26
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
27
 
28
- # Translate the prompt from Russian to English if necessary
29
- prompt = GoogleTranslator(source='id', target='en').translate(prompt)
30
- print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
31
 
32
- # Add some extra flair to the prompt
33
- prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
34
- print(f'\033[1mGeneration {key}:\033[0m {prompt}')
35
 
36
  # Prepare the payload for the API call, including width and height
37
  payload = {
 
13
  API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
14
  API_TOKEN = os.getenv("HF_READ_TOKEN")
15
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
16
+ timeout = 90000000
17
 
18
  # Function to query the API and return the generated image
19
  def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
 
25
  API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN")])
26
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
27
 
 
 
 
28
 
 
 
 
29
 
30
  # Prepare the payload for the API call, including width and height
31
  payload = {