Yaron Koresh commited on
Commit
0a6ec1f
·
verified ·
1 Parent(s): 77dabde

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(duration=120)
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=100,
72
  guidance_scale=10.0
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
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=1024,
69
+ width=1024,
70
  negative_prompt="",
71
+ num_inference_steps=50,
72
  guidance_scale=10.0
73
  ).images[0].save(name)
74
  return name