Spaces:
Running
on
L4
Running
on
L4
Alexander Becker
commited on
Commit
·
ea13900
1
Parent(s):
9b63d83
Bugfix
Browse files
app.py
CHANGED
|
@@ -15,7 +15,7 @@ from super_resolve import process
|
|
| 15 |
|
| 16 |
REPO_ID_EDSR = "prs-eth/thera-edsr-pro"
|
| 17 |
REPO_ID_RDN = "prs-eth/thera-rdn-pro"
|
| 18 |
-
MAX_SIZE = os.getenv('THERA_DEMO_CROP', 10_000)
|
| 19 |
|
| 20 |
print(f"JAX devices: {jax.devices()}")
|
| 21 |
print(f"JAX device type: {jax.devices()[0].device_kind}")
|
|
|
|
| 15 |
|
| 16 |
REPO_ID_EDSR = "prs-eth/thera-edsr-pro"
|
| 17 |
REPO_ID_RDN = "prs-eth/thera-rdn-pro"
|
| 18 |
+
MAX_SIZE = int(os.getenv('THERA_DEMO_CROP', 10_000))
|
| 19 |
|
| 20 |
print(f"JAX devices: {jax.devices()}")
|
| 21 |
print(f"JAX device type: {jax.devices()[0].device_kind}")
|