Spaces:
Running
on
Zero
Running
on
Zero
Upload 6 files
Browse files
feifeilib/feifeifluxapi.py
CHANGED
@@ -7,10 +7,10 @@ from feifeilib.feifeisharpened import feifeisharpened
|
|
7 |
|
8 |
client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
|
9 |
client.headers["x-use-cache"] = "0"
|
10 |
-
MAX_SEED = np.iinfo(np.int32).
|
11 |
|
12 |
-
def feifeifluxapi(randomize_seed=False,seed=random.randint(0, MAX_SEED),
|
13 |
-
prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
|
14 |
image = client.text_to_image(
|
15 |
prompt=prompt,
|
16 |
height=height,
|
@@ -18,5 +18,5 @@ def feifeifluxapi(randomize_seed=False,seed=random.randint(0, MAX_SEED),prompt,q
|
|
18 |
guidance_scale=guidance_scale
|
19 |
)
|
20 |
if sharpened_select:
|
21 |
-
feifeisharpened(image,num_strength)
|
22 |
-
return image
|
|
|
7 |
|
8 |
client = InferenceClient("black-forest-labs/FLUX.1-dev", token=os.getenv('HF_TOKEN'))
|
9 |
client.headers["x-use-cache"] = "0"
|
10 |
+
MAX_SEED = np.iinfo(np.int32).max
|
11 |
|
12 |
+
def feifeifluxapi(prompt, randomize_seed=False, seed=random.randint(0, MAX_SEED), quality_select=False, sharpened_select=False, styles_Radio=["(None)"], FooocusExpansion_select=False, height=1152, width=896, guidance_scale=3.5, num_strength=0.35):
|
13 |
+
prompt, generator = feifeiprompt(randomize_seed, seed, prompt, quality_select, styles_Radio, FooocusExpansion_select)
|
14 |
image = client.text_to_image(
|
15 |
prompt=prompt,
|
16 |
height=height,
|
|
|
18 |
guidance_scale=guidance_scale
|
19 |
)
|
20 |
if sharpened_select:
|
21 |
+
feifeisharpened(image, num_strength)
|
22 |
+
return image
|