Update app.py

app.py

@@ -111,8 +111,9 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
         height = height,
         generator = generator
     )
+    image=result.images[0].to("cpu")
     
-    return result.images[0]
+    return image
 
 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}
 """