lulavc commited on
Commit
93ff589
·
verified ·
1 Parent(s): 3cacffc

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -929,7 +929,7 @@ def transform_with_polish(input_image: Optional[Image.Image], prompt: str, style
929
  final_display = polished_display if polished_display else full_prompt
930
  yield image, final_display, used_seed
931
 
932
- @spaces.GPU(duration=60)
933
  def generate(full_prompt: str, polished_display: str, ratio: str, steps: int, seed: int, randomize: bool, progress=gr.Progress(track_tqdm=True)) -> Tuple[Optional[Image.Image], int]:
934
  """Generate image from text prompt."""
935
  if randomize:
@@ -958,7 +958,7 @@ def generate(full_prompt: str, polished_display: str, ratio: str, steps: int, se
958
  logger.error(f"Generation failed: {type(e).__name__}: {str(e)}")
959
  return None, seed
960
 
961
- @spaces.GPU(duration=45)
962
  def transform(input_image: Optional[Image.Image], full_prompt: str, polished_display: str, strength: float, steps: int, seed: int, randomize: bool, progress=gr.Progress(track_tqdm=True)) -> Tuple[Optional[Image.Image], int]:
963
  """Transform image using prompt guidance."""
964
  if input_image is None:
@@ -1005,7 +1005,7 @@ def transform(input_image: Optional[Image.Image], full_prompt: str, polished_dis
1005
  # These functions expose all parameters directly for MCP server compatibility
1006
  # =============================================================================
1007
 
1008
- @spaces.GPU(duration=60)
1009
  def mcp_generate(prompt: str, style: str = "None", ratio: str = "1:1 Square (1024x1024)",
1010
  steps: int = 9, seed: int = 42, randomize: bool = True) -> Tuple[Optional[Image.Image], int]:
1011
  """MCP-friendly image generation. Takes prompt directly and handles polish internally."""
@@ -1038,7 +1038,7 @@ def mcp_generate(prompt: str, style: str = "None", ratio: str = "1:1 Square (102
1038
  logger.error(f"MCP Generate failed: {type(e).__name__}: {str(e)}")
1039
  return None, seed
1040
 
1041
- @spaces.GPU(duration=45)
1042
  def mcp_transform(image: Optional[Image.Image], prompt: str, style: str = "None",
1043
  strength: float = 0.6, steps: int = 9, seed: int = 42,
1044
  randomize: bool = True) -> Tuple[Optional[Image.Image], int]:
 
929
  final_display = polished_display if polished_display else full_prompt
930
  yield image, final_display, used_seed
931
 
932
+ @spaces.GPU(duration=120)
933
  def generate(full_prompt: str, polished_display: str, ratio: str, steps: int, seed: int, randomize: bool, progress=gr.Progress(track_tqdm=True)) -> Tuple[Optional[Image.Image], int]:
934
  """Generate image from text prompt."""
935
  if randomize:
 
958
  logger.error(f"Generation failed: {type(e).__name__}: {str(e)}")
959
  return None, seed
960
 
961
+ @spaces.GPU(duration=90)
962
  def transform(input_image: Optional[Image.Image], full_prompt: str, polished_display: str, strength: float, steps: int, seed: int, randomize: bool, progress=gr.Progress(track_tqdm=True)) -> Tuple[Optional[Image.Image], int]:
963
  """Transform image using prompt guidance."""
964
  if input_image is None:
 
1005
  # These functions expose all parameters directly for MCP server compatibility
1006
  # =============================================================================
1007
 
1008
+ @spaces.GPU(duration=120)
1009
  def mcp_generate(prompt: str, style: str = "None", ratio: str = "1:1 Square (1024x1024)",
1010
  steps: int = 9, seed: int = 42, randomize: bool = True) -> Tuple[Optional[Image.Image], int]:
1011
  """MCP-friendly image generation. Takes prompt directly and handles polish internally."""
 
1038
  logger.error(f"MCP Generate failed: {type(e).__name__}: {str(e)}")
1039
  return None, seed
1040
 
1041
+ @spaces.GPU(duration=90)
1042
  def mcp_transform(image: Optional[Image.Image], prompt: str, style: str = "None",
1043
  strength: float = 0.6, steps: int = 9, seed: int = 42,
1044
  randomize: bool = True) -> Tuple[Optional[Image.Image], int]: