Update app.py
Browse files
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)
|