Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ML endpoints #1

Open
wants to merge 27 commits into
base: main
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
71657c3
YOLO implementation
DiegoPino May 6, 2024
f05d332
Unit vector. If i am doing this right or not? Math from 1998 for me
DiegoPino May 6, 2024
a01bc9f
Allow Yolo model to be set via an ENV
DiegoPino May 6, 2024
ea0a041
Much better output. All JSON, no weird strings.
DiegoPino May 6, 2024
c00ac91
Adds mobilenetv3 embedder (size 1024) via mediapipe (tensorflow/google)
DiegoPino May 6, 2024
9e3b0e1
adds Bert/sentence model loader and embedder
DiegoPino May 6, 2024
f99f5b2
return to 6400 for nlpserver.py
DiegoPino May 14, 2024
8dd8af9
Ignore models when committing
DiegoPino May 14, 2024
1023cdd
Smooth out the code
DiegoPino May 22, 2024
f8f97cd
Keep making the same mistakes. Gosh. No more alpha releases. let's gi…
DiegoPino May 22, 2024
70c6150
Nop. probs are always None
DiegoPino May 23, 2024
b700b5e
Trying with norm L2, because dot_product between same vector is not 1…
DiegoPino May 23, 2024
94c2be1
Literally need to stop doing this
DiegoPino May 23, 2024
4b8a675
torch Tensor stuff condition + empirical confidence of at least 0.3
DiegoPino May 23, 2024
f85591b
Fix mobilenet "capabilities" check/key
DiegoPino May 23, 2024
0676246
WE ship with the large model now
DiegoPino May 23, 2024
6fe9d47
Because not convinced that L2 (even if correct) handles size/coverage…
DiegoPino May 23, 2024
da1a18f
Python driving me crazy
DiegoPino May 23, 2024
33cc504
more argument checking
DiegoPino May 23, 2024
85d9f57
So distracted. Single argument not a list
DiegoPino May 23, 2024
0946d78
First pass on using BGE-M3 (sentence transformer)
DiegoPino May 24, 2024
f5e07bf
Adds InsightFace. Not an easy one
DiegoPino May 25, 2024
de16332
Reserved Dimensions of Image
DiegoPino May 28, 2024
695bef6
print the Embedding Size on the logs (for future data-scientist delig…
DiegoPino Jun 24, 2024
a823aeb
Keras is driving me nuts
DiegoPino Jul 11, 2024
e7cb7ba
WIP ViT endpoint. Also Polyglot is having issues so commenting out th…
DiegoPino Jan 15, 2025
4ebe3f2
Try to convert via a buffer past Greyscale to RGB
DiegoPino Jan 25, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion nlpserver.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
# configurations
#app.config['var1'] = 'test'
app.config["YOLO_MODEL_NAME"] = "yolov8m.pt"
app.config["MOBILENET_MODEL_NAME"] = "mobilenet_v3_small.tflite"
app.config["MOBILENET_MODEL_NAME"] = "mobilenet_v3_large.tflite"

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👍

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Really a matter of space/taste/ The small one really had little inference capabilities. But I will defer to you on this

app.config["EFFICIENTDET_DETECT_MODEL_NAME"] = "efficientdet_lite2.tflite"
app.config["MOBILENET_DETECT_MODEL_NAME"] = "ssd_mobilenet_v2.tflite"
for variable, value in os.environ.items():
Expand Down