geshang commited on
Commit
60971a4
·
verified ·
1 Parent(s): 1f86a8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -233,8 +233,7 @@ def visualize_masks_on_image(
233
  blended = cv2.addWeighted(image_np, 1 - alpha, color_mask, alpha, 0)
234
  return PILImage.fromarray(blended)
235
 
236
- @torch.no_grad()
237
- @spaces.GPU(duration=60)
238
  def run_pipeline(image: PILImage.Image, prompt: str):
239
  if not model or not processor:
240
  return "Models not loaded. Please check logs.", None
 
233
  blended = cv2.addWeighted(image_np, 1 - alpha, color_mask, alpha, 0)
234
  return PILImage.fromarray(blended)
235
 
236
+
 
237
  def run_pipeline(image: PILImage.Image, prompt: str):
238
  if not model or not processor:
239
  return "Models not loaded. Please check logs.", None