Update app.py

app.py

@@ -14,14 +14,14 @@ import torch
 
 #######################################
 #Bild nach dem eingegebenen prompt erzeugen - mit Pipeline
-def erzeuge(prompt):
+def erzeuge_1(prompt):
     pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
     pipeline.to("cuda")
     return pipeline(prompt).images[0]
 
 ########################################
 #Bild erzeugen - nich über Pipeline sondern mit mehr Einstellungsmöglichkeiten
-def erzeuge_komplex(prompt):    
+def erzeuge(prompt):    
     scheduler = DDPMScheduler.from_pretrained("google/ddpm-cat-256")
     model = UNet2DModel.from_pretrained("google/ddpm-cat-256").to("cuda")
     scheduler.set_timesteps(50)
@@ -60,6 +60,7 @@ with gr.Blocks() as demo:
         ).style(columns=[2], rows=[2], object_fit="contain", height="auto")
 
     btn.click(erzeuge, inputs=[text], outputs=[gallery])
+    submitBtn.click(erzeuge, inputs=[text], outputs=[gallery])
 
 if __name__ == "__main__":
     demo.launch()