alexkueck commited on
Commit
1ce879b
·
1 Parent(s): 4eeec90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +48 -1
app.py CHANGED
@@ -1,3 +1,50 @@
1
  import gradio as gr
 
 
2
 
3
- gr.Interface.load("models/stabilityai/stable-diffusion-2").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from diffusers import DiffusionPipeline
3
+ import random
4
 
5
+
6
+ #########################
7
+ #Alternativ erzeugen
8
+ #gr.Interface.load("models/stabilityai/stable-diffusion-2").launch()
9
+
10
+ pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
11
+
12
+ def erzeuge():
13
+ pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2")
14
+ images = [
15
+ (random.choice(
16
+ [
17
+ "https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=387&q=80",
18
+ "https://images.unsplash.com/photo-1554151228-14d9def656e4?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=386&q=80",
19
+ "https://images.unsplash.com/photo-1542909168-82c3e7fdca5c?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxzZWFyY2h8MXx8aHVtYW4lMjBmYWNlfGVufDB8fDB8fA%3D%3D&w=1000&q=80",
20
+ "https://images.unsplash.com/photo-1546456073-92b9f0a8d413?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=387&q=80",
21
+ "https://images.unsplash.com/photo-1601412436009-d964bd02edbc?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=464&q=80",
22
+ ]
23
+ ), f"label {i}" if i != 0 else "label" * 50)
24
+ for i in range(3)
25
+ ]
26
+ return images
27
+
28
+
29
+
30
+ with gr.Blocks() as demo:
31
+ with gr.Column(variant="panel"):
32
+ with gr.Row(variant="compact"):
33
+ text = gr.Textbox(
34
+ label="Deine Beschreibung:",
35
+ show_label=False,
36
+ max_lines=1,
37
+ placeholder="Bildbeschreibung",
38
+ ).style(
39
+ container=False,
40
+ )
41
+ btn = gr.Button("Bild Erzeugung:").style(full_width=False)
42
+
43
+ gallery = gr.Gallery(
44
+ label="Erzeugte Bilder", show_label=False, elem_id="gallery"
45
+ ).style(columns=[2], rows=[2], object_fit="contain", height="auto")
46
+
47
+ btn.click(erzeuge, None, gallery)
48
+
49
+ if __name__ == "__main__":
50
+ demo.launch()