zhiweili
commited on
Commit
·
e84a93e
1
Parent(s):
b28f15d
add gpu time
Browse files- app_enhance.py +1 -1
app_enhance.py
CHANGED
@@ -40,7 +40,7 @@ half = True if torch.cuda.is_available() else False
|
|
40 |
upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
41 |
|
42 |
|
43 |
-
@spaces.GPU(duration=
|
44 |
def enhance_image(
|
45 |
input_image: Image,
|
46 |
scale: int,
|
|
|
40 |
upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
41 |
|
42 |
|
43 |
+
@spaces.GPU(duration=15)
|
44 |
def enhance_image(
|
45 |
input_image: Image,
|
46 |
scale: int,
|