Vvaann commited on
Commit
d0c5619
·
verified ·
1 Parent(s): f7aadc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def inference(input_img, transparancy = 0.5, target_layer_number = -1):
55
 
56
  return classes[prediction[0].item(),visualization,confidences]
57
 
58
- demo = gr.Inference(
59
  inference,
60
  inputs = [
61
  gr.Image(width=256,height=256,label="input image"),
 
55
 
56
  return classes[prediction[0].item(),visualization,confidences]
57
 
58
+ demo = gr.Interface(
59
  inference,
60
  inputs = [
61
  gr.Image(width=256,height=256,label="input image"),