Spaces:
Sleeping
Sleeping
ahmedfaiyaz
commited on
Commit
•
e6d6963
1
Parent(s):
0dfd0ac
added support for all the models.
Browse files
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
|