Module: Rager::ImageGen

Extended by:
T::Sig
Defined in:
lib/rager/image_gen.rb,
lib/rager/image_gen/options.rb,
lib/rager/image_gen/providers/abstract.rb,
lib/rager/image_gen/providers/replicate.rb

Defined Under Namespace

Modules: Providers Classes: Options

Class Method Summary collapse

Class Method Details

.get_provider(key) ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/rager/image_gen.rb', line 22

def self.get_provider(key)
  case key.downcase
  when "replicate"
    Rager::ImageGen::Providers::Replicate.new
  else
    raise Rager::Errors::UnknownProviderError.new(Rager::Operation::ImageGen, key)
  end
end

.image_gen(prompt, options = Rager::ImageGen::Options.new) ⇒ Object 



16
17
18
19
 
# File 'lib/rager/image_gen.rb', line 16

def self.image_gen(prompt, options = Rager::ImageGen::Options.new)
  provider = get_provider(options.provider)
  provider.image_gen(prompt, options)
end