tbdatasci commited on
Commit
aade435
·
1 Parent(s): 92e00d0

Trying to get this running locally

Browse files
Files changed (1) hide show
  1. app.py +29 -29
app.py CHANGED
@@ -1,40 +1,40 @@
1
- # import gradio as gr
2
- # from diffusers import DiffusionPipeline
3
 
4
- # # get_completion = pipeline("image-to-text",model="nlpconnect/vit-gpt2-image-captioning")
5
 
6
- # # pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
7
- # pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
8
 
9
- # # def summarize(input):
10
- # # output = get_completion(input)
11
- # # return output[0]['generated_text']
12
 
13
- # # def captioner(image):
14
- # # result = get_completion(image)
15
- # # return result[0]['generated_text']
16
 
17
- # def generate(prompt):
18
- # return pipeline(prompt).images[0]
19
 
20
- # gr.close_all()
21
- # demo = gr.Interface(fn=generate,
22
- # inputs=[gr.Textbox(label="Your prompt")],
23
- # outputs=[gr.Image(label="Result")],
24
- # title="Image Generation with Stable Diffusion",
25
- # description="Generate any image with Stable Diffusion",
26
- # allow_flagging="never",
27
- # examples=["the spirit of a tamagotchi wandering in the city of Vienna","a mecha robot in a favela"])
28
 
29
- # demo.launch()
30
 
31
 
32
- import gradio as gr
33
 
34
- gr.close_all()
35
- demo = gr.load(name="models/stabilityai/stable-diffusion-xl-base-1.0",
36
- title='PicassoBot Large',
37
- enable_queue=True,
38
- description='Because paint splatters are so last century')
39
 
40
- demo.launch(enable_queue=True)
 
1
+ import gradio as gr
2
+ from diffusers import DiffusionPipeline
3
 
4
+ # get_completion = pipeline("image-to-text",model="nlpconnect/vit-gpt2-image-captioning")
5
 
6
+ # pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
7
+ pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
8
 
9
+ # def summarize(input):
10
+ # output = get_completion(input)
11
+ # return output[0]['generated_text']
12
 
13
+ # def captioner(image):
14
+ # result = get_completion(image)
15
+ # return result[0]['generated_text']
16
 
17
+ def generate(prompt):
18
+ return pipeline(prompt).images[0]
19
 
20
+ gr.close_all()
21
+ demo = gr.Interface(fn=generate,
22
+ inputs=[gr.Textbox(label="Your prompt")],
23
+ outputs=[gr.Image(label="Result")],
24
+ title="Image Generation with Stable Diffusion",
25
+ description="Generate any image with Stable Diffusion",
26
+ allow_flagging="never",
27
+ examples=["the spirit of a tamagotchi wandering in the city of Vienna","a mecha robot in a favela"])
28
 
29
+ demo.launch()
30
 
31
 
32
+ # import gradio as gr
33
 
34
+ # gr.close_all()
35
+ # demo = gr.load(name="models/stabilityai/stable-diffusion-xl-base-1.0",
36
+ # title='PicassoBot Large',
37
+ # enable_queue=True,
38
+ # description='Because paint splatters are so last century')
39
 
40
+ # demo.launch(enable_queue=True)