DGSpitzer commited on
Commit
3220d0e
1 Parent(s): 4fd5a25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -37,6 +37,10 @@ tips = {"en": "Tips: The input text will be translated into Chinese for generati
37
 
38
  count = 0
39
 
 
 
 
 
40
  model_id2 = "riffusion/riffusion-model-v1"
41
  pipe2 = StableDiffusionPipeline.from_pretrained(model_id2, torch_dtype=torch.float16)
42
  pipe2 = pipe2.to("cuda")
@@ -71,9 +75,6 @@ def get_result(text_prompts, style_indx):
71
  style = style_list[style_indx]
72
  prompt = text_prompts + "," + style
73
 
74
- model_id = "runwayml/stable-diffusion-v1-5"
75
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
76
- pipe = pipe.to("cuda")
77
 
78
  image_output = pipe(prompt).images[0]
79
 
 
37
 
38
  count = 0
39
 
40
+ model_id = "runwayml/stable-diffusion-v1-5"
41
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
42
+ pipe = pipe.to("cuda")
43
+
44
  model_id2 = "riffusion/riffusion-model-v1"
45
  pipe2 = StableDiffusionPipeline.from_pretrained(model_id2, torch_dtype=torch.float16)
46
  pipe2 = pipe2.to("cuda")
 
75
  style = style_list[style_indx]
76
  prompt = text_prompts + "," + style
77
 
 
 
 
78
 
79
  image_output = pipe(prompt).images[0]
80