Update app.py

app.py

@@ -28,7 +28,20 @@ a = np.ones((100,100,3),np.uint8)
 a[:,:,0]=255
 st.image(a)
 
+import requests
+
+API_URL = "https://api-inference.huggingface.co/models/nvidia/segformer-b5-finetuned-ade-640-640"
+headers = {"Authorization": "Bearer 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"}
+
+def query(filename):
+    with open(filename, "rb") as f:
+        data = f.read()
+    response = requests.request("POST", API_URL, headers=headers, data=data)
+    return json.loads(response.content.decode("utf-8"))
 
+output = query("cats.jpg")
+st.write(output.shape)
+"""
 from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 from PIL import Image
 import requests
@@ -43,3 +56,4 @@ 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)
+"""