MrDrmm commited on
Commit
0e84cf5
·
verified ·
1 Parent(s): 853d6bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -4,7 +4,16 @@ 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
+ import { HfInference } from "@huggingface/inference";
8
 
9
+ const inference = new HfInference(HF_TOKEN);
10
+ await inference.textToImage({
11
+ model: "stabilityai/stable-diffusion-2",
12
+ inputs: "award winning high resolution photo of a giant tortoise/((ladybird)) hybrid, [trending on artstation]",
13
+ parameters: {
14
+ negative_prompt: "blurry",
15
+ },
16
+ });
17
  max_images = 8
18
  MAX_SEED = 2**32-1
19
  load_models(models)