Update app.py

app.py

@@ -167,11 +167,12 @@ def respond(
         mask = pred_pil.resize(image_size)
         image.putalpha(mask)
         print('remver success')
+        
         try:
-            return str(convert_image_to_base64(image))
+            yield str(convert_image_to_base64(image))
         except Exception as e:
             print(e)
-            return image
+            yield image
         
 
     except Exception as e: