Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -73,9 +73,9 @@ _PrC.__getattribute__ = _getattr
|
|
73 |
|
74 |
model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
|
75 |
|
76 |
-
# Florence-2
|
77 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
78 |
-
scheduler =
|
79 |
model_repo,
|
80 |
subfolder="scheduler",
|
81 |
torch_dtype=torch.float16,
|
@@ -98,7 +98,7 @@ florence_model.to("cpu")
|
|
98 |
florence_model.eval()
|
99 |
florence_processor = AutoProcessor.from_pretrained(LOCAL_FLORENCE, trust_remote_code=True)
|
100 |
|
101 |
-
# Stable Diffusion TurboX
|
102 |
|
103 |
diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
|
104 |
|
@@ -116,11 +116,9 @@ pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, sub
|
|
116 |
MAX_SEED = 2**31 - 1
|
117 |
|
118 |
def pseudo_translate_to_korean_style(en_prompt: str) -> str:
|
119 |
-
# ๋ฒ์ญ ์์ด ์คํ์ผ ์ ์ฉ
|
120 |
return f"Cartoon styled {en_prompt} handsome or pretty people"
|
121 |
|
122 |
def generate_prompt(image):
|
123 |
-
"""์ด๋ฏธ์ง โ ์์ด ์ค๋ช
โ ํ๊ตญ์ด ํ๋กฌํํธ ์คํ์ผ๋ก ๋ณํ"""
|
124 |
if not isinstance(image, Image.Image):
|
125 |
image = Image.fromarray(image)
|
126 |
|
|
|
73 |
|
74 |
model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
|
75 |
|
76 |
+
# Florence-2
|
77 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
78 |
+
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(
|
79 |
model_repo,
|
80 |
subfolder="scheduler",
|
81 |
torch_dtype=torch.float16,
|
|
|
98 |
florence_model.eval()
|
99 |
florence_processor = AutoProcessor.from_pretrained(LOCAL_FLORENCE, trust_remote_code=True)
|
100 |
|
101 |
+
# Stable Diffusion TurboX
|
102 |
|
103 |
diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
|
104 |
|
|
|
116 |
MAX_SEED = 2**31 - 1
|
117 |
|
118 |
def pseudo_translate_to_korean_style(en_prompt: str) -> str:
|
|
|
119 |
return f"Cartoon styled {en_prompt} handsome or pretty people"
|
120 |
|
121 |
def generate_prompt(image):
|
|
|
122 |
if not isinstance(image, Image.Image):
|
123 |
image = Image.fromarray(image)
|
124 |
|