Update app.py

app.py

@@ -13,12 +13,12 @@ import numpy as np
 
 #######################################
 #Bild nach dem eingegebenen prompt erzeugen - mit Pipeline
-def erzeuge_1(prompt):
-    return pipeline(prompt).images[0]
+def erzeuge(prompt):
+    return pipeline(prompt).images
 
 ########################################
 #Bild erzeugen - nich über Pipeline sondern mit mehr Einstellungsmöglichkeiten
-def erzeuge(prompt):    
+def erzeuge_komplex(prompt):    
     scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
     model = UNet2DModel.from_pretrained("google/ddpm-cat-256").to("cuda")
     scheduler.set_timesteps(50)