Spaces:
Running
on
Zero
Running
on
Zero
Upload app.py with huggingface_hub
Browse files
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=
|
| 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=
|
| 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=
|
| 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=
|
| 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]:
|