hugging-face 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 053432b377155f8768a2e25521b62c0fc5fdbb3a619f45875c346c5f3c126e81
-  data.tar.gz: d8551f8263129317a2692e5b6d3994617c1ae66285ce21a78a1796a2ad5f9f85
+  metadata.gz: 93e213102e56d8e86de856912a1963f36e4b0a0d60f21c8a1f8d99b23a0f47d7
+  data.tar.gz: 877bd5a93e54ccbcb82943c19e4ac889657686aa656b0bd3bb911732b1d5d6ff
 SHA512:
-  metadata.gz: 933fad128d5ad66370e0e24271498cc9730e0f71e9ed920cf5a3107f8971bb5382586d1bc3d887b4d490b0fbbb8f0498e9b503dab4a669469bc1349824e023a3
-  data.tar.gz: 0654e009acdbea3097f00f13f7c2b4389679e52b4b15db629ca2703c4267f2e27dcbfe866ee644c9cb5c23d56a5f5960d37a18334c7d0a21ca55815941aec281
+  metadata.gz: ee534a915fa831884e42519f178ae709a08c201d79976d97b52319dd5322fc979c3dfe711a3e2210b0d53c0132490a089ccd2ca7c73730919f586582be0dc884
+  data.tar.gz: c4df146833ea986a86d9b9c060f594535b7bdf558dd9d28396dd100869c4a899dc4ddad1bdd2eba1ff8a09610ee4cada09a271adda9f4d6e58d21c5181c925a6

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    hugging-face (0.1.0)
+    hugging-face (0.2.0)
       faraday (~> 2.7)
 
 GEM

data/README.md

@@ -28,7 +28,7 @@ Instantiate a HuggigFace Inference API client:
 client = HuggingFace::InferenceApi.new(api_key: ENV['HUGGING_FACE_API_KEY'])
 ```
 
-Questiion answering:
+Question answering:
 
 ```ruby
 client.question_answering(

data/lib/hugging_face/base_api.rb

@@ -15,7 +15,7 @@ module HuggingFace
 
     private
 
-    def connection(url)
+    def build_connection(url)
       Faraday.new(url, headers: @headers)
     end
 

data/lib/hugging_face/inference_api.rb

@@ -1,11 +1,15 @@
 module HuggingFace
   class InferenceApi < BaseApi
     HOST = "https://api-inference.huggingface.co"
-    MAX_RETRY = 20
 
+    # Retry connecting to the model for 1 minute
+    MAX_RETRY = 60
+
+    # Deafult models that can be overriden by 'model' param
     QUESTION_ANSWERING_MODEL = 'distilbert-base-cased-distilled-squad'
     SUMMARIZATION_MODEL = "sshleifer/distilbart-xsum-12-6"
     GENERATION_MODEL = "distilgpt2"
+    EMBEDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
 
     def call(input:, model:)
       request(connection: connection(model), input: input)
@@ -14,21 +18,29 @@ module HuggingFace
     def question_answering(question:, context:, model: QUESTION_ANSWERING_MODEL)
       input = { question: question, context: context }
 
-      request(connection: connection(model), input: input)
+      request connection: connection(model), input: input
     end
 
     def summarization(input:, model: SUMMARIZATION_MODEL)
-      request(connection: connection(model), input: { inputs: input })
+      request connection: connection(model), input: { inputs: input }
     end
 
     def text_generation(input:, model: GENERATION_MODEL)
-      request(connection: connection(model), input: { inputs: input })
+      request connection: connection(model), input: { inputs: input }
+    end
+
+    def embedding(input:)
+      request connection: connection(EMBEDING_MODEL), input: { inputs: input }
     end
 
     private
 
     def connection(model)
-      super "#{HOST}/models/#{model}"
+      if model == EMBEDING_MODEL
+        build_connection "#{HOST}/pipeline/feature-extraction/#{model}"
+      else
+        build_connection "#{HOST}/models/#{model}"
+      end
     end
 
     def request(connection:, input:)

data/lib/hugging_face/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module HuggingFace
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hugging-face
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Alex Chaplinsky