Sushrut98 commited on
Commit
c21a5e4
·
verified ·
1 Parent(s): c5e4496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -75,7 +75,7 @@ description = "An app to predict wafer Defects from the following 9 defects.[Cen
75
  # Create Gradio interface
76
  demo = gr.Interface(
77
  fn=predict,
78
- inputs=gr.Image(),
79
  outputs=[
80
  gr.Label(num_top_classes=5, label="Predictions"),
81
  gr.Image(label="Defects"),
@@ -86,4 +86,4 @@ demo = gr.Interface(
86
  )
87
 
88
  # Launch the app!
89
- demo.launch()
 
75
  # Create Gradio interface
76
  demo = gr.Interface(
77
  fn=predict,
78
+ inputs=gr.Image(type="pil"),
79
  outputs=[
80
  gr.Label(num_top_classes=5, label="Predictions"),
81
  gr.Image(label="Defects"),
 
86
  )
87
 
88
  # Launch the app!
89
+ demo.launch(share=True)