smileycutie😻 commited on
Commit
8aee7b6
·
verified ·
1 Parent(s): d34aa96

Update client.py

Browse files
Files changed (1) hide show
  1. client.py +3 -1
client.py CHANGED
@@ -1,6 +1,7 @@
1
  import random
2
  import time
3
  from io import BytesIO
 
4
  from urllib.parse import quote
5
 
6
  import requests
@@ -40,7 +41,8 @@ class Inference:
40
  "enhance": "true",
41
  "nologo": "true",
42
  "private": "true",
43
- "referrer": "cherry-ghosts"
 
44
  }
45
  image = self._fetch(url, params=params)
46
  if image:
 
1
  import random
2
  import time
3
  from io import BytesIO
4
+ from os import getenv
5
  from urllib.parse import quote
6
 
7
  import requests
 
41
  "enhance": "true",
42
  "nologo": "true",
43
  "private": "true",
44
+ "referrer": "cherry-ghosts",
45
+ "token": getenv("TOKEN")
46
  }
47
  image = self._fetch(url, params=params)
48
  if image: