Spaces:
Sleeping
Sleeping
roman-bachmann
commited on
Commit
•
d41a2a0
1
Parent(s):
3222587
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ def img_from_path(img_path: str):
|
|
67 |
img = rgb_transform.postprocess(img_pil).unsqueeze(0)
|
68 |
return img
|
69 |
|
70 |
-
@spaces.GPU
|
71 |
def infer(img_path, seed=0, randomize_seed=False, target_modalities=None, top_p=0.8, top_k=0.0):
|
72 |
if randomize_seed:
|
73 |
seed = None
|
|
|
67 |
img = rgb_transform.postprocess(img_pil).unsqueeze(0)
|
68 |
return img
|
69 |
|
70 |
+
@spaces.GPU(duration=80)
|
71 |
def infer(img_path, seed=0, randomize_seed=False, target_modalities=None, top_p=0.8, top_k=0.0):
|
72 |
if randomize_seed:
|
73 |
seed = None
|