Alexander Becker commited on
Commit
ea13900
·
1 Parent(s): 9b63d83
Files changed (1) hide show
  1. app.py +1 -1
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}")