hugging-face 0.3.1 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9b9591169889f3b463224dade2fa11e63bba1733d6e09ebb814e9e9e1d5bad14
-  data.tar.gz: 521110a79c57838d68aff92db3212358bcda75484b55a78c92ea46272fbc41d2
+  metadata.gz: 8b1fd1e517f21466be454d55dcc88e0344acaa6fd91900335615f632b734595e
+  data.tar.gz: c865080f1350cc0e056f0ff190b6c99373cc32a287bb9c0e558ce777456ca057
 SHA512:
-  metadata.gz: 9940b1ec90fdb03631d3318f1f12bf6431132166d97adbb9e3e962f6acf7e2150982da478129cee4c2d2bb4bdb973f9dda2b7e837c4722ad2d9b09d122205172
-  data.tar.gz: 4d9076507fcc8b3f6829aeb34d462fda23a317b8b2009a94be73cceb33b255f871ece5a0d0b41a832b743b4c26e7aa6af9d589f0eb3c112b96d4760ac56434b4
+  metadata.gz: 803618ef477824cafbf3b5d37e76d3bcfcc6b9cfd90632e55df733f9b7f14d7ef6e4d5403ce9f1828f6459ee32b511783c2d8f3217caf7ba3ee87123f0372754
+  data.tar.gz: 62cfb9bb3f0f6905f9963d7997b7deecabdda75a0771f8625604bde1dd7c883ab76bc43e811355b99f26428d0ae84397deb1a5741824fa3c32fcd0e80fa6cc1c

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    hugging-face (0.3.1)
+    hugging-face (0.3.3)
       faraday (~> 1.0)
 
 GEM

data/lib/hugging_face/base_api.rb

@@ -1,3 +1,4 @@
+require 'json'
 require 'logger'
 require 'faraday'
 

data/lib/hugging_face/inference_api.rb

@@ -29,8 +29,8 @@ module HuggingFace
       request connection: connection(model), input: { inputs: input }
     end
 
-    def embedding(input:)
-      request connection: connection(EMBEDING_MODEL), input: { inputs: input }
+    def embedding(input:, model: EMBEDING_MODEL)
+      request connection: connection(model), input: { inputs: input }
     end
 
     private

data/lib/hugging_face/version.rb

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

metadata

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