multimodalart HF Staff commited on
Commit
0c2491d
·
verified ·
1 Parent(s): 2a9e571

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=120)
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=150)
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,