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)
|