doevent commited on
Commit
c8067b8
1 Parent(s): 290a326

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model8 = RealESRGAN(device, scale=8)
12
  model8.load_weights('weights/RealESRGAN_x8.pth', download=True)
13
 
14
 
15
- def inference(image: Image, size: str) -> Image:
16
  if size == '2x':
17
  result = model2.predict(image.convert('RGB'))
18
  elif size == '4x':
 
12
  model8.load_weights('weights/RealESRGAN_x8.pth', download=True)
13
 
14
 
15
+ def inference(image, size):
16
  if size == '2x':
17
  result = model2.predict(image.convert('RGB'))
18
  elif size == '4x':