yiw commited on
Commit
2785bbf
·
1 Parent(s): 914dad5

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
  classifier = pipeline("image-classification", model="google/vit-base-patch16-224")
5
  def get_img_from_url(url):
6
  return Image.open(requests.get(url, stream=True).raw)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
  from PIL import Image
4
+ import requests
5
+
6
  classifier = pipeline("image-classification", model="google/vit-base-patch16-224")
7
  def get_img_from_url(url):
8
  return Image.open(requests.get(url, stream=True).raw)