ar08 commited on
Commit
ccfbed4
·
verified ·
1 Parent(s): 259924e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from diffusers import AutoPipelineForImage2Image
4
- from PIL import Image
5
  from typing import List
6
  #StableDiffusionImg2ImgPipeline
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -13,7 +12,6 @@ pipe = AutoPipelineForImage2Image.from_pretrained(
13
  safety_checker=None
14
  ).to(device)
15
 
16
- pipe.load_lora_weights("openfree/flux-chatgpt-ghibli-lora")
17
  pipe.enable_attention_slicing()
18
 
19
  styles = {
 
1
  import gradio as gr
2
  import torch
3
+ from diffusers import StableDiffusionImg2ImgPipeline
 
4
  from typing import List
5
  #StableDiffusionImg2ImgPipeline
6
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
12
  safety_checker=None
13
  ).to(device)
14
 
 
15
  pipe.enable_attention_slicing()
16
 
17
  styles = {