Update app.py

app.py

@@ -41,21 +41,5 @@ def query(filename):
     return json.loads(response.content.decode("utf-8"))
 
 output = query("pers.png")
-st.write(output)
-#st.write(output[0])
-"""
-from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
-from PIL import Image
-import requests
-
-feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b5-finetuned-ade-512-512")
-model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b5-finetuned-ade-512-512")
-
-url = "http://images.cocodataset.org/val2017/000000039769.jpg"
-image = Image.open(requests.get(url, stream=True).raw)
-
-inputs = feature_extractor(images=image, return_tensors="pt")
-outputs = model(**inputs)
-logits = outputs.logits  # shape (batch_size, num_labels, height/4, width/4)
-st.write(logits.shape)
-""" 
+#st.write(output)
+st.write(output[0])