cutiee82 commited on
Commit
7b74fb2
·
verified ·
1 Parent(s): 75d68af

Update client.py

Browse files
Files changed (1) hide show
  1. client.py +2 -7
client.py CHANGED
@@ -6,11 +6,9 @@ from urllib.parse import quote
6
  import requests
7
  from PIL import Image
8
 
9
- MAX_SEED = 2 ** 31 - 1
10
-
11
  class Inference:
12
  def __init__(self):
13
- self.sesion = requests.Session()
14
 
15
  def _fetch(self, *args, retries=5, **kwargs):
16
  for retry in range(retries + 1):
@@ -29,13 +27,10 @@ class Inference:
29
  def __call__(
30
  self,
31
  prompt,
32
- seed=None,
33
- randomize_seed=True,
34
  width=1024,
35
  height=1024
36
  ):
37
- if seed is None or randomize_seed:
38
- seed = random.randint(0, MAX_SEED)
39
  url = f"https://image.pollinations.ai/p/{prompt}"
40
  params = {
41
  "prompt": prompt,
 
6
  import requests
7
  from PIL import Image
8
 
 
 
9
  class Inference:
10
  def __init__(self):
11
+ self.session = requests.Session()
12
 
13
  def _fetch(self, *args, retries=5, **kwargs):
14
  for retry in range(retries + 1):
 
27
  def __call__(
28
  self,
29
  prompt,
30
+ seed=42,
 
31
  width=1024,
32
  height=1024
33
  ):
 
 
34
  url = f"https://image.pollinations.ai/p/{prompt}"
35
  params = {
36
  "prompt": prompt,