Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 54 |
MAX_SEED = np.iinfo(np.int32).max
|
| 55 |
|
| 56 |
|
| 57 |
-
@spaces.GPU
|
| 58 |
def infer(
|
| 59 |
content_ref,
|
| 60 |
style_ref,
|
|
|
|
| 54 |
MAX_SEED = np.iinfo(np.int32).max
|
| 55 |
|
| 56 |
|
| 57 |
+
@spaces.GPU(size="xlarge")
|
| 58 |
def infer(
|
| 59 |
content_ref,
|
| 60 |
style_ref,
|