Leeps commited on
Commit
0a1f600
1 Parent(s): d4ed719

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. index.py +4 -6
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": 25,
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": 512,
65
- "height": 512,
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": 25,
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
  }