alfredplpl commited on
Commit
1051b3b
1 Parent(s): 4dc2497

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -110,9 +110,8 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
110
  height = height,
111
  generator = generator
112
  )
113
- image=result.images[0].to("cpu")
114
 
115
- return image
116
 
117
  css = """.main-div div{display:inline-flex;align-items:center;gap:.8rem;font-size:1.75rem}.main-div div h1{font-weight:900;margin-bottom:7px}.main-div p{margin-bottom:10px;font-size:94%}a{text-decoration:underline}.tabs{margin-top:0;margin-bottom:0}#gallery{min-height:20rem}
118
  """
 
110
  height = height,
111
  generator = generator
112
  )
 
113
 
114
+ return result.images[0]
115
 
116
  css = """.main-div div{display:inline-flex;align-items:center;gap:.8rem;font-size:1.75rem}.main-div div h1{font-weight:900;margin-bottom:7px}.main-div p{margin-bottom:10px;font-size:94%}a{text-decoration:underline}.tabs{margin-top:0;margin-bottom:0}#gallery{min-height:20rem}
117
  """