3koozy commited on
Commit
dea36c5
·
1 Parent(s): 4076aa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import segmentation_models_pytorch as smp
3
  import torch
 
4
 
5
  #load our pytorch model:
6
  model = smp.Unet(
 
1
  import gradio as gr
2
  import segmentation_models_pytorch as smp
3
  import torch
4
+ import PIL as Image
5
 
6
  #load our pytorch model:
7
  model = smp.Unet(