florentgbelidji HF staff commited on
Commit
de1f3eb
1 Parent(s): f5f8dd8

Create new file

Browse files
Files changed (1) hide show
  1. README.md +52 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - feature-extraction
4
+ - endpoints-template
5
+ license: bsd-3-clause
6
+ library_name: generic
7
+ ---
8
+ # Fork of [salesforce/BLIP](https://github.com/salesforce/BLIP) for a `feature-extraction` task on 🤗Inference endpoint.
9
+ This repository implements a `custom` task for `feature-extraction` for 🤗 Inference Endpoints. The code for the customized pipeline is in the [pipeline.py](https://huggingface.co/florentgbelidji/blip-embeddings/blob/main/pipeline.py).
10
+ To use deploy this model a an Inference Endpoint you have to select `Custom` as task to use the `pipeline.py` file. -> _double check if it is selected_
11
+ ### expected Request payload
12
+ ```json
13
+ {
14
+ "image": "/9j/4AAQSkZJRgABAQEBLAEsAAD/2wBDAAMCAgICAgMC....", // base64 image as bytes
15
+ }
16
+ ```
17
+ below is an example on how to run a request using Python and `requests`.
18
+ ## Run Request
19
+ 1. prepare an image.
20
+ ```bash
21
+ !wget https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
22
+ ```
23
+ 2.run request
24
+ ```python
25
+ import json
26
+ from typing import List
27
+ import requests as r
28
+ import base64
29
+ ENDPOINT_URL = ""
30
+ HF_TOKEN = ""
31
+ def predict(path_to_image: str = None):
32
+ with open(path_to_image, "rb") as i:
33
+ b64 = base64.b64encode(i.read())
34
+ payload = {"inputs": {"image": b64.decode("utf-8")}}
35
+ response = r.post(
36
+ ENDPOINT_URL, headers={"Authorization": f"Bearer {HF_TOKEN}"}, json=payload
37
+ )
38
+ return response.json()
39
+ prediction = predict(
40
+ path_to_image="palace.jpg"
41
+ )
42
+ ```
43
+ expected output
44
+ ```python
45
+ {'feature_vector': [0.016450975090265274,
46
+ -0.5551009774208069,
47
+ 0.39800673723220825,
48
+ -0.6809228658676147,
49
+ 2.053842782974243,
50
+ -0.4712907075881958,...]
51
+ }
52
+ ```