Omnibus commited on
Commit
3705f9b
1 Parent(s): 6f78e57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -71,14 +71,16 @@ def star_fn(inp):
71
  uid=uuid.uuid4()
72
  output=proc2.send_it(inp,5,1)
73
  outp=Image.open(output[0])
74
- outrs = outp.resize((24,24))
75
- print("resized")
76
- outbg=rm(outrs)
 
 
77
  #outbg.save(f"{uid}_star.png")
78
  #out = os.path.abspath(f"{uid}_star.png")
79
  #out_url = f'https://omnibus-game-test.hf.space/file={out}'
80
-
81
- return outbg,outbg
82
 
83
  with gr.Blocks() as app:
84
  with gr.Row():
 
71
  uid=uuid.uuid4()
72
  output=proc2.send_it(inp,5,1)
73
  outp=Image.open(output[0])
74
+
75
+ print("removing background")
76
+ outbg=rm(outp)
77
+ print("resizing")
78
+ out = outbg.resize((24,24))
79
  #outbg.save(f"{uid}_star.png")
80
  #out = os.path.abspath(f"{uid}_star.png")
81
  #out_url = f'https://omnibus-game-test.hf.space/file={out}'
82
+
83
+ return out,out
84
 
85
  with gr.Blocks() as app:
86
  with gr.Row():