Vijish commited on
Commit
1eb089e
1 Parent(s): 3c9073a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -75,6 +75,7 @@ learn=load_learner(path, 'popd.pkl')
75
  def predict(image):
76
  img_fast = open_image(image)
77
  a = PIL.Image.open(image).convert('RGB')
 
78
  p,img_hr,b = learn.predict(img_fast)
79
  x = np.minimum(np.maximum(image2np(img_hr.data*255), 0), 255).astype(np.uint8)
80
  img = PIL.Image.fromarray(x).convert('RGB')
 
75
  def predict(image):
76
  img_fast = open_image(image)
77
  a = PIL.Image.open(image).convert('RGB')
78
+ st.image(a, caption='Input')
79
  p,img_hr,b = learn.predict(img_fast)
80
  x = np.minimum(np.maximum(image2np(img_hr.data*255), 0), 255).astype(np.uint8)
81
  img = PIL.Image.fromarray(x).convert('RGB')