fahmiaziz98 commited on
Commit
f417379
·
1 Parent(s): cced668

first commit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline, AutoImageProcessor
5
  from PIL import Image
6
  from utils import download_model_from_s3
7
 
8
- model_path = {
9
  "TinyBert Sentiment Analysis": "ml-models/tinybert-sentiment-analysis/",
10
  "TinyBert Disaster Classification": "ml-models/tinybert-disaster-tweet/",
11
  "VIT Pose Classification": "ml-models/vit-human-pose-classification/"
@@ -76,4 +76,4 @@ if model_choice == "VIT Pose Classification":
76
  if predict:
77
  with st.spinner("Predicting..."):
78
  output = pipe(image)
79
- st.write(output)
 
5
  from PIL import Image
6
  from utils import download_model_from_s3
7
 
8
+ model_paths = {
9
  "TinyBert Sentiment Analysis": "ml-models/tinybert-sentiment-analysis/",
10
  "TinyBert Disaster Classification": "ml-models/tinybert-disaster-tweet/",
11
  "VIT Pose Classification": "ml-models/vit-human-pose-classification/"
 
76
  if predict:
77
  with st.spinner("Predicting..."):
78
  output = pipe(image)
79
+ st.write(output)