tbdatasci commited on
Commit
664cd28
·
1 Parent(s): a9c9ec3

Changed back to API

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
+ description='Because paint splatters are so last century')
 
38
 
39
+ demo.queue(concurrency_count=1)
40
+ demo.launch()