Update app.py

app.py

@@ -48,17 +48,25 @@ def load_b_lora_to_unet(pipe, content_lora_model_id: str = '', style_lora_model_
             raise type(e)(f'failed to load_b_lora_to_unet, due to: {e}')
 
 def main(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+        
+    generator = torch.Generator().manual_seed(seed)
+    
     content_B_LoRA_path = ''
-    style_B_LoRA_path = 'fffiloni/b_lora_trained_test_7'
+    style_B_LoRA_path = 'fffiloni/b_lora_tardi_4'
     content_alpha,style_alpha = 1,1.1
 
     load_b_lora_to_unet(pipeline, content_B_LoRA_path, style_B_LoRA_path, content_alpha, style_alpha)
-    prompt = 'An eagle in [v42] style'
+    prompt = prompt
     image = pipeline(
         prompt,
-        generator=torch.Generator(device="cuda").manual_seed(48), 
-        num_images_per_prompt=1
-    ).images[0].resize((512,512))
+        generator=generator, 
+        num_images_per_prompt=1,
+        width = width, 
+        height = height,
+    ).images[0])
     
     pipeline.unload_lora_weights()