Spaces:

xinghaochen
/

tinysam

Sleeping

xinghaochen commited on Jan 6

Commit

507e5e7

•

1 Parent(s): de08842

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -100,7 +100,7 @@ def on_image_upload(
     session_state['image_with_prompt'] = copy.deepcopy(image)
     print("Image changed")
     nd_image = np.array(image)
-    session_state['feature'] = None #predictor.set_image(nd_image)
     return image, session_state

     session_state['image_with_prompt'] = copy.deepcopy(image)
     print("Image changed")
     nd_image = np.array(image)
+    session_state['feature'] = predictor.set_image(nd_image)
     return image, session_state