RMBG_2_Studio-Kaggle.ipynb

{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.10.14","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"},"kaggle":{"accelerator":"nvidiaTeslaT4","dataSources":[],"dockerImageVersionId":30787,"isInternetEnabled":true,"language":"python","sourceType":"notebook","isGpuEnabled":true}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"code","source":"%cd /kaggle/working/\n!git clone https://github.com/zachysaur/RMBG-2-Studio.git\n%cd RMBG-2-Studio/app\n!pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118\n!pip install -r requirements.txt","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"outputs":[],"execution_count":null},{"cell_type":"code","source":"!npm install -g localtunnel\n\nimport subprocess\nimport threading\nimport time\nimport socket\nimport urllib.request\n\ndef iframe_thread(port):\n  while True:\n      time.sleep(0.5)\n      sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)\n      result = sock.connect_ex(('127.0.0.1', port))\n      if result == 0:\n        break\n      sock.close()\n  print(\"\\nComfyUI finished loading, trying to launch localtunnel (if it gets stuck here localtunnel is having issues)\\n\")\n\n  print(\"The password/enpoint ip for localtunnel is:\", urllib.request.urlopen('https://ipv4.icanhazip.com').read().decode('utf8').strip(\"\\n\"))\n  p = subprocess.Popen([\"lt\", \"--port\", \"{}\".format(port)], stdout=subprocess.PIPE)\n  for line in p.stdout:\n    print(line.decode(), end='')\n\n\nthreading.Thread(target=iframe_thread, daemon=True, args=(7860,)).start()\n\n!python /kaggle/working/RMBG-2-Studio/app/app.py","metadata":{"trusted":true},"outputs":[],"execution_count":null}]}