Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,18 @@ from multit2i import (load_models, infer_fn, infer_rand_fn, save_gallery,
|
|
4 |
change_model, warm_model, get_model_info_md, loaded_models,
|
5 |
get_positive_prefix, get_positive_suffix, get_negative_prefix, get_negative_suffix,
|
6 |
get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
|
|
|
7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
max_images = 8
|
9 |
MAX_SEED = 2**32-1
|
10 |
load_models(models)
|
|
|
4 |
change_model, warm_model, get_model_info_md, loaded_models,
|
5 |
get_positive_prefix, get_positive_suffix, get_negative_prefix, get_negative_suffix,
|
6 |
get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
|
7 |
+
from huggingface_hub import InferenceClient
|
8 |
|
9 |
+
client = InferenceClient(
|
10 |
+
provider="hf-inference",
|
11 |
+
api_key="hf_xxxxxxxxxxxxxxxxxxxxxxxx"
|
12 |
+
)
|
13 |
+
|
14 |
+
# output is a PIL.Image object
|
15 |
+
image = client.text_to_image(
|
16 |
+
"Astronaut riding a horse",
|
17 |
+
model="hugovntr/flux-schnell-realism"
|
18 |
+
)
|
19 |
max_images = 8
|
20 |
MAX_SEED = 2**32-1
|
21 |
load_models(models)
|