Update app.py

app.py

@@ -5,9 +5,12 @@ from PIL import Image
 import uuid
 import random
 import bg
+from rembg import remove as rm
 proc1=diff
 proc2=diff 
-bg1=gr.Interface.load("spaces/Omnibus/fast-bg-remove-mod")
+
+
+#bg1=gr.Interface.load("spaces/Omnibus/fast-bg-remove-mod")
 #proc2=gr.Interface.load("spaces/Omnibus/fast-diff")
 
 sky = "https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/sky.png"
@@ -65,31 +68,17 @@ def star_fn(inp):
     rand = random.randint(1,200)
     for i in range(rand):
         inp+=" "
-    #dif = gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base", live=True),
-    #dif = models[0]
     uid=uuid.uuid4()
-    #output=proc2(inp,5,1)
     output=proc2.send_it(inp,5,1)
-    print(output)
-    #output=dif(inp)
-    #in_url=f'https://omnibus-fast-diff.hf.space/file={output[0]}'
-        
     outp=Image.open(output[0])
-    #outp.save(f"{uid}_pre_star.png")
-    
-    #out = os.path.abspath(f"{uid}_pre_star.png")
-    #out_pre_url = f'https://omnibus-game-test.hf.space/file={out}'     
-    #print(out_pre_url)
-    outp = outp.resize((24,24))
-    
-    outp=bg.rem_bg(outp)
+    outrs = outp.resize((24,24))
+    print("resized")
+    outbg=rm(outrs)
+    #outbg.save(f"{uid}_star.png")
+    #out = os.path.abspath(f"{uid}_star.png")
+    #out_url = f'https://omnibus-game-test.hf.space/file={out}'    
 
-    #bg_im=Image.open(outp_bg)
-    #print(f'outp_bg:: {outp_bg}')
-    outp.save(f"{uid}_star.png")
-    out = os.path.abspath(f"{uid}_star.png")
-    out_url = f'https://omnibus-game-test.hf.space/file={out}'    
-    return out_url,out_url
+    return outbg,outbg
 
 with gr.Blocks() as app:
     with gr.Row():