jhoppanne commited on
Commit
d8fc58c
1 Parent(s): f6f09af

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  import numpy as np
5
  from PIL import Image
6
  age_classifier = pipeline("image-classification", model="nateraw/vit-age-classifier")
7
- emotion_classifier = pipeline("image-classification", model="jhoppanne/Image-Emotion-Classification")
8
  def pred_age_emotion(input_image):
9
  if isinstance(input_image,np.ndarray):
10
  img = Image.fromarray(input_image)
 
4
  import numpy as np
5
  from PIL import Image
6
  age_classifier = pipeline("image-classification", model="nateraw/vit-age-classifier")
7
+ emotion_classifier = pipeline("image-classification", model="jhoppanne/Emotion-Image-Classification-V2")
8
  def pred_age_emotion(input_image):
9
  if isinstance(input_image,np.ndarray):
10
  img = Image.fromarray(input_image)
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  import numpy as np
5
  from PIL import Image
6
  age_classifier = pipeline("image-classification", model="nateraw/vit-age-classifier")
7
- emotion_classifier = pipeline("image-classification", model="jhoppanne/Image-Emotion-Classification")
8
  def pred_age_emotion(input_image):
9
  if isinstance(input_image,np.ndarray):
10
  img = Image.fromarray(input_image)
 
4
  import numpy as np
5
  from PIL import Image
6
  age_classifier = pipeline("image-classification", model="nateraw/vit-age-classifier")
7
+ emotion_classifier = pipeline("image-classification", model="jhoppanne/Emotion-Image-Classification-V2")
8
  def pred_age_emotion(input_image):
9
  if isinstance(input_image,np.ndarray):
10
  img = Image.fromarray(input_image)