Spaces:
Running on Zero
Running on Zero
Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ model = AutoModel.from_pretrained(MODEL_ID, config=config, torch_dtype=torch.bfl
|
|
| 102 |
print("[startup] model ready.")
|
| 103 |
|
| 104 |
|
| 105 |
-
@spaces.GPU(duration=
|
| 106 |
def generate(
|
| 107 |
prompt: str,
|
| 108 |
aspect_ratio: str = "1:1",
|
|
@@ -144,7 +144,7 @@ def generate(
|
|
| 144 |
return _to_pil(tensor)[0], seed
|
| 145 |
|
| 146 |
|
| 147 |
-
@spaces.GPU(duration=
|
| 148 |
def edit(
|
| 149 |
image: Image.Image,
|
| 150 |
prompt: str,
|
|
|
|
| 102 |
print("[startup] model ready.")
|
| 103 |
|
| 104 |
|
| 105 |
+
@spaces.GPU(duration=100)
|
| 106 |
def generate(
|
| 107 |
prompt: str,
|
| 108 |
aspect_ratio: str = "1:1",
|
|
|
|
| 144 |
return _to_pil(tensor)[0], seed
|
| 145 |
|
| 146 |
|
| 147 |
+
@spaces.GPU(duration=110)
|
| 148 |
def edit(
|
| 149 |
image: Image.Image,
|
| 150 |
prompt: str,
|