Update app.py
Browse files
app.py
CHANGED
@@ -58,8 +58,8 @@ def generate_image(prompt, model, negative_prompt, sampling_method, sampling_ste
|
|
58 |
model = AutoModelForConditionalGeneration.from_pretrained("vqgan/vqgan_imagenet_f16_1024_clip_cutout")
|
59 |
tokenizer = AutoTokenizer.from_pretrained("vqgan/vqgan_imagenet_f16_1024_clip_cutout")
|
60 |
elif model == "VQGAN+CLIP-RandomizedCutout":
|
61 |
-
|
62 |
-
|
63 |
|
64 |
prompt = f"{prompt} {negative_prompt}"
|
65 |
|
|
|
58 |
model = AutoModelForConditionalGeneration.from_pretrained("vqgan/vqgan_imagenet_f16_1024_clip_cutout")
|
59 |
tokenizer = AutoTokenizer.from_pretrained("vqgan/vqgan_imagenet_f16_1024_clip_cutout")
|
60 |
elif model == "VQGAN+CLIP-RandomizedCutout":
|
61 |
+
model = AutoModelForConditionalGeneration.from_pretrained("vqgan/vqgan_imagenet_f16_1024_clip_randomized_cutout")
|
62 |
+
tokenizer = AutoTokenizer.from_pretrained("vqgan/vqgan_imagenet_f16_1024_clip_randomized_cutout")
|
63 |
|
64 |
prompt = f"{prompt} {negative_prompt}"
|
65 |
|