JOY-Huang commited on
Commit
15874c5
1 Parent(s): 61a0f40
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- print(os.listdir('.'))
3
  import torch
4
  import random
5
  import numpy as np
@@ -31,7 +31,7 @@ transform = transforms.Compose([
31
 
32
  device = "cuda" if torch.cuda.is_available() else "cpu"
33
  sdxl_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"
34
- instantir_repo_id = "instantx/instantir"
35
  dinov2_repo_id = "facebook/dinov2-large"
36
 
37
  if torch.cuda.is_available():
 
1
  import os
2
+ print(os.listdir("InstantX/InstantIR"))
3
  import torch
4
  import random
5
  import numpy as np
 
31
 
32
  device = "cuda" if torch.cuda.is_available() else "cpu"
33
  sdxl_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"
34
+ instantir_repo_id = "InstantX/InstantIR"
35
  dinov2_repo_id = "facebook/dinov2-large"
36
 
37
  if torch.cuda.is_available():