Hugging Face's logo

Spaces:
amirza
/
draw_me_a_sheep_heb
Runtime error

App Files Community
Amir Zait commited on
Commit
8439859
•
1 Parent(s): e8b13db

name fix

Browse files
Files changed (1) hide show
  1. image_generator.py +1 -1
image_generator.py CHANGED
@@ -20,7 +20,7 @@ model = DalleBart.from_pretrained(DALLE_MODEL, revision=DALLE_COMMIT_ID)
20
  vqgan = VQModel.from_pretrained(VQGAN_REPO, revision=VQGAN_COMMIT_ID)
21
  processor = DalleBartProcessor.from_pretrained(DALLE_MODEL, revision=DALLE_COMMIT_ID)
22
 
23
- def get_image(text):
24
  tokenized_prompt = processor([text])
25
 
26
  gen_top_k = None
 
20
  vqgan = VQModel.from_pretrained(VQGAN_REPO, revision=VQGAN_COMMIT_ID)
21
  processor = DalleBartProcessor.from_pretrained(DALLE_MODEL, revision=DALLE_COMMIT_ID)
22
 
23
+ def generate_image(text):
24
  tokenized_prompt = processor([text])
25
 
26
  gen_top_k = None