File size: 975 Bytes
876b664
 
 
 
3fe250d
 
 
876b664
 
 
a04510d
876b664
 
3fe250d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
876b664
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import time
import utilities as u
import card_generator as card
from PIL import Image
import replicate
from pathlib import Path


start_time = time.time()
temp_image_path = "./image_temp/"



def preview_and_generate_image(num_images,sd_prompt, user_input_template, item):    
    # Replicate is expecting a path
    card_template = Path(user_input_template[0][0])
    print(card_template)
    img_start = time.time()   
    output=replicate.run(
            "drakosfire/card-generator-v1:4e46bb44e6444d104b58e34a6d2ab66833aa84ba0dfe84ae57ef63d8e15da467",
            input={
            "item":item,
            "sd_prompt":sd_prompt,
            "num_images":num_images,
            "input_template":card_template

        }
    )
    
    
    img_time = time.time() - img_start
    img_its = 35/img_time
    print(f"image gen time = {img_time} and {img_its} it/s")
    
    total_time = time.time() - start_time
    print(total_time)

    return output