rdjarbeng commited on
Commit
88a3311
·
1 Parent(s): 0abf88f

add logging of rembg version

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  from rembg import remove, new_session
3
  from PIL import Image
4
  import numpy as np
@@ -7,7 +8,8 @@ import time
7
 
8
  # Set up logging
9
  logging.basicConfig(level=logging.INFO)
10
-
 
11
  # Define model options with separate names and descriptions
12
  MODEL_OPTIONS = {
13
  "": "Select a model",
@@ -118,4 +120,5 @@ iface = gr.Interface(
118
  )
119
 
120
  if __name__ == "__main__":
 
121
  iface.launch()
 
1
  import gradio as gr
2
+ import rembg
3
  from rembg import remove, new_session
4
  from PIL import Image
5
  import numpy as np
 
8
 
9
  # Set up logging
10
  logging.basicConfig(level=logging.INFO)
11
+ # Log the version of rembg
12
+ logging.info(f"rembg version: {rembg.__version__}")
13
  # Define model options with separate names and descriptions
14
  MODEL_OPTIONS = {
15
  "": "Select a model",
 
120
  )
121
 
122
  if __name__ == "__main__":
123
+
124
  iface.launch()