Tapanat commited on
Commit
dc8ba2d
·
1 Parent(s): d60d001

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -1,25 +1,24 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
  from PIL import Image
4
- #import req
5
  import requests
6
 
7
  classifier = pipeline("image-classification", model="google/vit-base-patch16-224")
 
8
  def get_img_from_url(url):
9
  return Image.open(requests.get(url, stream=True).raw)
10
 
11
  def main():
12
- st.title("image-classification")
13
 
14
  with st.form("image"):
15
-
16
- url = st.text_input("URL to some image", "https://images.livemint.com/img/2022/08/01/600x338/Cat-andriyko-podilnyk-RCfi7vgJjUY-unsplash_1659328989095_1659328998370_1659328998370.jpg")
17
  img = get_img_from_url(url)
18
- # clicked==True only when the button is clicked
19
- clicked = st.form_submit_button("Submit")
20
  if clicked:
21
- results = classifier(img)
22
- st.json(results)
23
 
24
  if __name__ == "__main__":
25
- main()
 
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
+
8
  def get_img_from_url(url):
9
  return Image.open(requests.get(url, stream=True).raw)
10
 
11
  def main():
12
+ st.title("Image Classification")
13
 
14
  with st.form("image"):
15
+ url = st.text_input("URL ของรูปภาพ", "https://images.livemint.com/img/2022/08/01/600x338/Cat-andriyko-podilnyk-RCfi7vgJjUY-unsplash_1659328989095_1659328998370_1659328998370.jpg")
 
16
  img = get_img_from_url(url)
17
+ clicked = st.form_submit_button("ส่งคำขอ")
18
+
19
  if clicked:
20
+ results = classifier(img)
21
+ st.json(results)
22
 
23
  if __name__ == "__main__":
24
+ main()