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