vikasdeep commited on
Commit
9494f59
·
verified ·
1 Parent(s): 7871e7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -4,7 +4,6 @@ import os
4
  # os.system('pip install --upgrade pip')
5
  # os.system('pip install --upgrade gradio')
6
 
7
-
8
  import gradio as gr
9
  import torch
10
  import torch.nn.functional as F
@@ -80,11 +79,9 @@ def predict(input_image: Image.Image):
80
 
81
  interface = gr.Interface(
82
  fn=predict,
83
- inputs=[
84
- gr.inputs.Image(label="Input Image", type="pil")
85
- ],
86
  outputs=[
87
- gr.outputs.Label(label="Class"),
88
- gr.outputs.Image(label="Face with Explainability", type="pil")
89
  ],
90
  ).launch(share=True)
 
4
  # os.system('pip install --upgrade pip')
5
  # os.system('pip install --upgrade gradio')
6
 
 
7
  import gradio as gr
8
  import torch
9
  import torch.nn.functional as F
 
79
 
80
  interface = gr.Interface(
81
  fn=predict,
82
+ inputs=gr.Image(label="Input Image", type="pil"),
 
 
83
  outputs=[
84
+ gr.Label(label="Class"),
85
+ gr.Image(label="Face with Explainability", type="pil")
86
  ],
87
  ).launch(share=True)