Hugging Face's logo

Spaces:
Kingoteam
/
Txt2img
Runtime error

App Files Community
Kingoteam commited on
Commit
0a635b0
·
verified ·
1 Parent(s): bafd4f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import numpy as np
7
  from PIL import Image
8
 
9
  # دانلود مدل‌ها
10
- DALLE_MODEL = "dalle-mini/dalle-mini/mega-1-fp16:latest" # یا dalle-mini/dalle-mini
11
- VQGAN_REPO = "dalle-mini/vqgan_imagenet_f16_16384"
12
 
13
  # بارگذاری پردازشگر و مدل
14
  processor = DalleBartProcessor.from_pretrained(DALLE_MODEL)
 
7
  from PIL import Image
8
 
9
  # دانلود مدل‌ها
10
+ DALLE_MODEL = "dalle-mini/dalle-mini" # یا dalle-mini/dalle-mini
11
+ VQGAN_REPO = "flax_model.msgpack"
12
 
13
  # بارگذاری پردازشگر و مدل
14
  processor = DalleBartProcessor.from_pretrained(DALLE_MODEL)