Geek7 commited on
Commit
fa42b05
·
verified ·
1 Parent(s): 90d6fa8

Update myapp.py

Browse files
Files changed (1) hide show
  1. myapp.py +1 -1
myapp.py CHANGED
@@ -10,7 +10,7 @@ CORS(myapp)
10
 
11
  device = torch.device('cpu')
12
  model = RealESRGAN(device, scale=2)
13
- model.load_weights('RealESRGAN_x2.pth', download=True)
14
 
15
  @myapp.route('/')
16
  def home():
 
10
 
11
  device = torch.device('cpu')
12
  model = RealESRGAN(device, scale=2)
13
+ model.load_weights('RealESRGAN_x2.pth', download=True,weights_only=True )
14
 
15
  @myapp.route('/')
16
  def home():