Rooni commited on
Commit
37e8f70
1 Parent(s): 3b8a061

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -8,12 +8,16 @@ from deep_translator import GoogleTranslator
8
  from langdetect import detect
9
 
10
  API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
11
- API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
12
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
13
-
14
  models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "PixArt XL 2.0", "Disney Cartoon", "CleanLinearMix", "Waifu 1.4"]
15
 
16
  def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
 
 
 
 
17
  language = detect(prompt)
18
  key = random.randint(0, 999)
19
 
 
8
  from langdetect import detect
9
 
10
  API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
11
+ API_TOKEN = os.getenv("HF_READ_TOKEN")
12
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
13
+ keys_list = json.loads(os.getenv("HF_READ_TOKEN"))
14
  models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjourney 4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit 2.5", "Realistic Vision 5.1", "Incursios 1.6", "Anime Detailer XL", "Vector Art XL", "epiCRealism", "PixelArt XL", "NewReality XL", "Anything 5.0", "PixArt XL 2.0", "Disney Cartoon", "CleanLinearMix", "Waifu 1.4"]
15
 
16
  def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, seed=None, sampler="DPM++ 2M Karras"):
17
+ if prompt == None:
18
+ return None
19
+
20
+ API_TOKEN = random.choice(keys_list)
21
  language = detect(prompt)
22
  key = random.randint(0, 999)
23