ahmedfaiyaz commited on
Commit
e6d6963
1 Parent(s): 0dfd0ac

added support for all the models.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -452,7 +452,7 @@ character_mappings_model_wise={
452
  def generate(modelname:str,input_text:str,batch_size:int,inference_steps:int):
453
  batch_size=int(batch_size)
454
  inference_steps=int(inference_steps)
455
- print(f"Generating image with label:{character_mappings[input_text]} batch size:{batch_size}")
456
  label=int(character_mappings_model_wise[current_model][input_text])
457
  pipeline.embedding=torch.tensor([label])
458
  generate_image=pipeline(batch_size=batch_size,num_inference_steps=inference_steps).images
 
452
  def generate(modelname:str,input_text:str,batch_size:int,inference_steps:int):
453
  batch_size=int(batch_size)
454
  inference_steps=int(inference_steps)
455
+ print(f"Generating image with label:{character_mappings[current_model][input_text]} batch size:{batch_size}")
456
  label=int(character_mappings_model_wise[current_model][input_text])
457
  pipeline.embedding=torch.tensor([label])
458
  generate_image=pipeline(batch_size=batch_size,num_inference_steps=inference_steps).images