Yaron Koresh commited on
Commit
9b8e50d
·
verified ·
1 Parent(s): 25cb026

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -58,17 +58,17 @@ def generate_random_string(length):
58
  characters = string.ascii_letters + string.digits
59
  return ''.join(random.choice(characters) for _ in range(length))
60
 
61
- @spaces.GPU
62
  def infer(prompt):
63
  name = generate_random_string(12)+".png"
64
  english_prompt = f'TRUE {translate(prompt,"english").upper()}:'
65
  print(f'Final prompt: {english_prompt}')
66
  image = pipe(
67
  english_prompt,
68
- height=768,
69
- width=768,
70
  negative_prompt="",
71
- num_inference_steps=120,
72
  guidance_scale=5
73
  ).images[0].save(name)
74
  return name
 
58
  characters = string.ascii_letters + string.digits
59
  return ''.join(random.choice(characters) for _ in range(length))
60
 
61
+ @spaces.GPU(20)
62
  def infer(prompt):
63
  name = generate_random_string(12)+".png"
64
  english_prompt = f'TRUE {translate(prompt,"english").upper()}:'
65
  print(f'Final prompt: {english_prompt}')
66
  image = pipe(
67
  english_prompt,
68
+ height=512,
69
+ width=512,
70
  negative_prompt="",
71
+ num_inference_steps=40,
72
  guidance_scale=5
73
  ).images[0].save(name)
74
  return name