MrDrmm commited on
Commit
24dd92f
·
verified ·
1 Parent(s): c14d123

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
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)