tsambo commited on
Commit
09c5df2
1 Parent(s): d7c0f42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
 
3
  from PIL import Image
 
4
 
5
  pipeline = pipeline(task = "image-classification", model = "microsoft/beit-base-patch16-224-pt22k-ft22k")
6
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
+ from transformers import BeitFeatureExtractor, BeitForImageClassification
4
  from PIL import Image
5
+ import requests
6
 
7
  pipeline = pipeline(task = "image-classification", model = "microsoft/beit-base-patch16-224-pt22k-ft22k")
8