Spaces:
Running
Running
Update client.py
Browse files
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.
|
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=
|
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,
|