nurindahpratiwi commited on
Commit
0627589
·
1 Parent(s): 3c54f75

change model and purpose

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -2,11 +2,12 @@ import streamlit as st
2
  from transformers import pipeline
3
  from PIL import Image
4
 
5
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
6
 
7
  st.title("Hot Dog? Or Not?")
8
 
9
- file_name = st.file_uploader("Upload a hot dog candidate image")
10
 
11
  if file_name is not None:
12
  col1, col2 = st.columns(2)
 
2
  from transformers import pipeline
3
  from PIL import Image
4
 
5
+ #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
6
+ pipeline = pipeline(task="image-classification", model="Rajaram1996/FacialEmoRecog")
7
 
8
  st.title("Hot Dog? Or Not?")
9
 
10
+ file_name = st.file_uploader("Upload a facial emote candidate image")
11
 
12
  if file_name is not None:
13
  col1, col2 = st.columns(2)