Textual Inversion training notebook only takes in remote images

The Textual Inversion training notebook in Google Colab only takes in remote images, instead of also being able to take in local images either uploaded from the desktop or linked from Google Drive, both of which are more common workflows.

urls = [
      "https://huggingface.co/datasets/valhalla/images/resolve/main/2.jpeg",
      "https://huggingface.co/datasets/valhalla/images/resolve/main/3.jpeg",
      "https://huggingface.co/datasets/valhalla/images/resolve/main/5.jpeg",
      "https://huggingface.co/datasets/valhalla/images/resolve/main/6.jpeg",
      ## You can add additional images here
      ]

import requests
import glob
from io import BytesIO

def download_image(url):
  try:
    response = requests.get(url)
  except:
    return None
  return Image.open(BytesIO(response.content)).convert("RGB")

images = list(filter(None,[download_image(url) for url in urls]))

The inputs to the image cell should be able to take in both remote or local images depending on if an external URL is detected.

Twitter discussion

[patrickvonplaten]

Gently ping here to update the notebook cc @patil-suraj

[patrickvonplaten]

Ping again @patil-suraj @williamberman @pcuenca