Update app.py

app.py

@@ -37,6 +37,10 @@ tips = {"en": "Tips: The input text will be translated into Chinese for generati
 
 count = 0
 
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
+
 model_id2 = "riffusion/riffusion-model-v1"
 pipe2 = StableDiffusionPipeline.from_pretrained(model_id2, torch_dtype=torch.float16)
 pipe2 = pipe2.to("cuda")
@@ -71,9 +75,6 @@ def get_result(text_prompts, style_indx):
     style = style_list[style_indx]
     prompt = text_prompts + "," + style
         
-    model_id = "runwayml/stable-diffusion-v1-5"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    pipe = pipe.to("cuda")
     
     image_output = pipe(prompt).images[0]