Futuretop commited on
Commit
f8215e3
ยท
verified ยท
1 Parent(s): fb29fe3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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 = EulerDiscreteScheduler.from_pretrained(
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