prithivMLmods commited on
Commit
ed6d187
·
verified ·
1 Parent(s): 834ff7b

update app [experimental]

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -147,7 +147,7 @@ def update_dimensions_on_upload(image):
147
 
148
  return new_width, new_height
149
 
150
- @spaces.GPU(duration=60)
151
  def infer(
152
  input_image,
153
  prompt,
@@ -215,7 +215,7 @@ def infer(
215
  gc.collect()
216
  torch.cuda.empty_cache()
217
 
218
- @spaces.GPU(duration=60)
219
  def infer_example(input_image, prompt, lora_adapter):
220
  if input_image is None:
221
  return None, 0
 
147
 
148
  return new_width, new_height
149
 
150
+ @spaces.GPU
151
  def infer(
152
  input_image,
153
  prompt,
 
215
  gc.collect()
216
  torch.cuda.empty_cache()
217
 
218
+ @spaces.GPU
219
  def infer_example(input_image, prompt, lora_adapter):
220
  if input_image is None:
221
  return None, 0