Yaron Koresh commited on
Commit
93ca623
·
verified ·
1 Parent(s): 6511739

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -483,11 +483,11 @@ torch.cuda.empty_cache()
483
 
484
  # functionality
485
 
486
- @spaces.GPU(duration=180)
487
  def hard_scaler(img):
488
  return upscaler(img)
489
 
490
- @spaces.GPU(duration=90)
491
  def easy_scaler(img):
492
  return upscaler(img)
493
 
@@ -498,7 +498,6 @@ def handle_upscaler(img):
498
  return hard_scaler(img)
499
  return easy_scaler(img)
500
 
501
- @spaces.GPU(duration=180)
502
  def upscaler(
503
  input_image: Image.Image,
504
  prompt: str = "Accurate, Highly Detailed, Realistic, Best Quality, Hyper-Realistic, Super-Realistic, Natural, Reasonable, Logical.",
@@ -1273,15 +1272,15 @@ def handle_generation(h,w,d):
1273
  else:
1274
  return hard_generation(h,w,d)
1275
 
1276
- @spaces.GPU(duration=90)
1277
  def easy_generation(h,w,d):
1278
  return generation(h,w,d)
1279
 
1280
- @spaces.GPU(duration=150)
1281
  def balanced_generation(h,w,d):
1282
  return generation(h,w,d)
1283
 
1284
- @spaces.GPU(duration=210)
1285
  def hard_generation(h,w,d):
1286
  return generation(h,w,d)
1287
 
 
483
 
484
  # functionality
485
 
486
+ @spaces.GPU(duration=300)
487
  def hard_scaler(img):
488
  return upscaler(img)
489
 
490
+ @spaces.GPU(duration=150)
491
  def easy_scaler(img):
492
  return upscaler(img)
493
 
 
498
  return hard_scaler(img)
499
  return easy_scaler(img)
500
 
 
501
  def upscaler(
502
  input_image: Image.Image,
503
  prompt: str = "Accurate, Highly Detailed, Realistic, Best Quality, Hyper-Realistic, Super-Realistic, Natural, Reasonable, Logical.",
 
1272
  else:
1273
  return hard_generation(h,w,d)
1274
 
1275
+ @spaces.GPU(duration=150)
1276
  def easy_generation(h,w,d):
1277
  return generation(h,w,d)
1278
 
1279
+ @spaces.GPU(duration=180)
1280
  def balanced_generation(h,w,d):
1281
  return generation(h,w,d)
1282
 
1283
+ @spaces.GPU(duration=240)
1284
  def hard_generation(h,w,d):
1285
  return generation(h,w,d)
1286