doevent commited on
Commit
8f30ed0
·
verified ·
1 Parent(s): 0f79ae4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,12 +13,12 @@ sr_pipe4x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=4)
13
  def inference(image, size):
14
  if image is None:
15
  raise gr.Error("Image not uploaded")
16
- r_image = Image.open(image)
17
 
18
  if size == '2x':
19
- result = sr_pipe2x(lr_image)
20
  else:
21
- result = sr_pipe4x(lr_image)
22
 
23
  print(f"Image size ({device}): {size} ... OK")
24
  return result
@@ -30,7 +30,7 @@ article = "<div style='text-align: center;'>Twitter <a href='https://twitter.com
30
 
31
 
32
  gr.Interface(inference,
33
- [gr.Image(type="filepath"),
34
  gr.Radio(['2x', '4x'],
35
  type="value",
36
  value='2x',
 
13
  def inference(image, size):
14
  if image is None:
15
  raise gr.Error("Image not uploaded")
16
+ # r_image = Image.open(image)
17
 
18
  if size == '2x':
19
+ result = sr_pipe2x(image)
20
  else:
21
+ result = sr_pipe4x(image)
22
 
23
  print(f"Image size ({device}): {size} ... OK")
24
  return result
 
30
 
31
 
32
  gr.Interface(inference,
33
+ [gr.Image(type="pil"),
34
  gr.Radio(['2x', '4x'],
35
  type="value",
36
  value='2x',