Adhir commited on
Commit
690cced
1 Parent(s): e1cb7b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -5,7 +5,11 @@ from transformers import pipeline
5
 
6
  #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
7
  #pipeline = pipeline(task="image-classification", model="hg2001/autotrain-animals-vs-humans2-37846100283")
8
- pipeline = pipeline("image-classification", model="lazyturtl/roomclassifier")
 
 
 
 
9
 
10
 
11
  def predict(image):
 
5
 
6
  #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
7
  #pipeline = pipeline(task="image-classification", model="hg2001/autotrain-animals-vs-humans2-37846100283")
8
+ #pipeline = pipeline("image-classification", model="lazyturtl/roomclassifier")
9
+
10
+ pipeline = pipeline("image-classification", model="dima806/facial_emotions_image_detection")
11
+
12
+
13
 
14
 
15
  def predict(image):