tombio commited on
Commit
36399f2
1 Parent(s): d8dabba

add mediapipe and numpy

Browse files
Files changed (2) hide show
  1. app.py +4 -1
  2. requirements.txt +2 -0
app.py CHANGED
@@ -20,6 +20,9 @@ from PIL import Image
20
  from fastai.vision.all import *
21
  import skimage
22
 
 
 
 
23
  from huggingface_hub import hf_hub_download
24
  ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt", cache_dir="/data/.cache")
25
  config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
@@ -188,4 +191,4 @@ gradio_interface = gradio.Interface(
188
  description="Turn portraits into a painting in the style of Flemish master Dirck Bouts",
189
  article="© iO Digital"
190
  )
191
- gradio_interface.launch()
 
20
  from fastai.vision.all import *
21
  import skimage
22
 
23
+ import mediapipe as mp
24
+ import numpy as np
25
+
26
  from huggingface_hub import hf_hub_download
27
  ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt", cache_dir="/data/.cache")
28
  config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
 
191
  description="Turn portraits into a painting in the style of Flemish master Dirck Bouts",
192
  article="© iO Digital"
193
  )
194
+ gradio_interface.queue().launch()
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
 
 
2
  torch==1.12.1
3
  torchvision==0.13.1
4
  albumentations==0.4.3
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
+ numpy
3
+ mediapipe
4
  torch==1.12.1
5
  torchvision==0.13.1
6
  albumentations==0.4.3