import streamlit as st from transformers import pipeline import matplotlib.pyplot as plt import requests from PIL import Image obj_model = pipeline("object-detection", model="facebook/detr-resnet-50") def get_img_from_url(url): return Image.open(requests.get(url, stream=True).raw) def main(): st.title("Object Detection") with st.form("text_field"): url = st.text_input("Enter an image URL", "https://images.unsplash.com/photo-1543852786-1cf6624b9987?ixlib=rb-1.2.1&ixid=MnwxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8&auto=format&fit=crop&w=987&q=80") st.write("*We recommend to use Unsplash website for browsing an image,[click here!](https://unsplash.com/)") img = get_img_from_url(url) # clicked==True only when the button is clicked clicked = st.form_submit_button("Submit") if clicked: results = obj_model(img) st.write("**——————————————————————**") st.write("**Input image:**") st.image(img) st.json(results) if __name__ == "__main__": main()