Andreas-w commited on
Commit
fdd320c
·
1 Parent(s): de77762

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -18,8 +18,9 @@ def funksjon(input_img):
18
 
19
  predictions = model.predict(input_img)
20
  #imgPred = input_img
21
-
22
- return predictions
 
23
 
24
 
25
  demo = gr.Interface(funksjon, gr.Image(), "image")
 
18
 
19
  predictions = model.predict(input_img)
20
  #imgPred = input_img
21
+ pil_image = Image.fromarray(predictions)
22
+ # Return the image
23
+ return pil_image
24
 
25
 
26
  demo = gr.Interface(funksjon, gr.Image(), "image")