Ashoka74 commited on
Commit
f97b08d
ยท
verified ยท
1 Parent(s): b248560

Update app_merged.py

Browse files
Files changed (1) hide show
  1. app_merged.py +2 -1
app_merged.py CHANGED
@@ -12,6 +12,7 @@ import spaces
12
  import argparse
13
  import random
14
 
 
15
  import os
16
  import math
17
  import gradio as gr
@@ -925,7 +926,7 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
925
  def extract_foreground(image):
926
  if image is None:
927
  return None, gr.update(visible=True), gr.update(visible=True)
928
- clear_memory()
929
  #logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
930
  #result, rgba = run_rmbg(image)
931
  result = run_rmbg(image)
 
12
  import argparse
13
  import random
14
 
15
+ import io
16
  import os
17
  import math
18
  import gradio as gr
 
926
  def extract_foreground(image):
927
  if image is None:
928
  return None, gr.update(visible=True), gr.update(visible=True)
929
+ # clear_memory()
930
  #logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
931
  #result, rgba = run_rmbg(image)
932
  result = run_rmbg(image)