Update app.py

app.py

@@ -1,6 +1,7 @@
 from transformers import DPTFeatureExtractor, DPTForSemanticSegmentation
 from PIL import Image
 import requests
+import streamlit as st
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
@@ -11,4 +12,5 @@ model = DPTForSemanticSegmentation.from_pretrained("Intel/dpt-large-ade")
 inputs = feature_extractor(images=image, return_tensors="pt")
 
 outputs = model(**inputs)
-logits = outputs.logits
+logits = outputs.logits
+st.json(logits)