Hugging Face's logo

Spaces:
Geek7
/
dnmgztxi
Runtime error

App Files Community
Geek7 commited on
Commit
2779248
·
verified ·
1 Parent(s): 59055d8

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +1 -1
myapp.py CHANGED
@@ -8,7 +8,7 @@ from RealESRGAN import RealESRGAN
8
  myapp = Flask(__name__)
9
  CORS(myapp)
10
 
11
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12
  model = RealESRGAN(device, scale=2)
13
  model.load_weights('RealESRGAN_x2.pth', download=True)
14
 
 
8
  myapp = Flask(__name__)
9
  CORS(myapp)
10
 
11
+ device = torch.device('cpu')
12
  model = RealESRGAN(device, scale=2)
13
  model.load_weights('RealESRGAN_x2.pth', download=True)
14