tbdatasci commited on
Commit
69e7707
·
1 Parent(s): b11353f

Change model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ 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)
 
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)