Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -483,11 +483,11 @@ torch.cuda.empty_cache()
|
|
483 |
|
484 |
# functionality
|
485 |
|
486 |
-
@spaces.GPU(duration=
|
487 |
def hard_scaler(img):
|
488 |
return upscaler(img)
|
489 |
|
490 |
-
@spaces.GPU(duration=
|
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=
|
1277 |
def easy_generation(h,w,d):
|
1278 |
return generation(h,w,d)
|
1279 |
|
1280 |
-
@spaces.GPU(duration=
|
1281 |
def balanced_generation(h,w,d):
|
1282 |
return generation(h,w,d)
|
1283 |
|
1284 |
-
@spaces.GPU(duration=
|
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 |
|