Update app.py

app.py

@@ -3,6 +3,11 @@ import gradio as gr
 import ast
 import requests
 
+import logging
+from rembg import new_session
+from cutter import remove, make_label
+from utils import *
+
 API_URL_INITIAL = "https://ysharma-playground-ai-exploration.hf.space/run/initial_dataframe"
 API_URL_NEXT10 = "https://ysharma-playground-ai-exploration.hf.space/run/next_10_rows"
 
@@ -21,6 +26,16 @@ models = [
     {"name": "Tron Legacy Diffusion", "url": "dallinmackay/Tron-Legacy-diffusion"},
 ]   
 
+
+####  REM-BG
+remove_bg_models = {
+    "TracerUniversalB7": "TracerUniversalB7",
+    "U2NET": "u2net",
+    "U2NET Human Seg": "u2net_human_seg",
+    "U2NET Cloth Seg": "u2net_cloth_seg"
+}
+###########
+
 text_gen = gr.Interface.load("spaces/daspartho/prompt-extend")
 
 current_model = models[0]