Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files
index.py
CHANGED
@@ -47,13 +47,11 @@ def image_classifier(prompt, starter_image, image_strength):
|
|
47 |
|
48 |
if starter_image is not None:
|
49 |
input = {
|
50 |
-
"width": 512,
|
51 |
-
"height": 512,
|
52 |
"prompt": prompt + " in the style of TOK",
|
53 |
"negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch",
|
54 |
#"refine": "expert_ensemble_refiner",
|
55 |
"apply_watermark": False,
|
56 |
-
"num_inference_steps":
|
57 |
"num_outputs": 3,
|
58 |
"lora_scale": .96,
|
59 |
"image": "data:image/jpeg;base64," + starter_image_base64,
|
@@ -61,13 +59,13 @@ def image_classifier(prompt, starter_image, image_strength):
|
|
61 |
}
|
62 |
else:
|
63 |
input = {
|
64 |
-
"width":
|
65 |
-
"height":
|
66 |
"prompt": prompt + " in the style of TOK",
|
67 |
"negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch",
|
68 |
#"refine": "expert_ensemble_refiner",
|
69 |
"apply_watermark": False,
|
70 |
-
"num_inference_steps":
|
71 |
"num_outputs": 3,
|
72 |
"lora_scale": .96,
|
73 |
}
|
|
|
47 |
|
48 |
if starter_image is not None:
|
49 |
input = {
|
|
|
|
|
50 |
"prompt": prompt + " in the style of TOK",
|
51 |
"negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch",
|
52 |
#"refine": "expert_ensemble_refiner",
|
53 |
"apply_watermark": False,
|
54 |
+
"num_inference_steps": 50,
|
55 |
"num_outputs": 3,
|
56 |
"lora_scale": .96,
|
57 |
"image": "data:image/jpeg;base64," + starter_image_base64,
|
|
|
59 |
}
|
60 |
else:
|
61 |
input = {
|
62 |
+
"width": 1024,
|
63 |
+
"height": 1024,
|
64 |
"prompt": prompt + " in the style of TOK",
|
65 |
"negative_prompt": "worst quality, low quality, illustration, 2d, painting, cartoons, sketch",
|
66 |
#"refine": "expert_ensemble_refiner",
|
67 |
"apply_watermark": False,
|
68 |
+
"num_inference_steps": 50,
|
69 |
"num_outputs": 3,
|
70 |
"lora_scale": .96,
|
71 |
}
|