Spaces:

chansung
/

co-write-with-llama2

Runtime error

chansung commited on Aug 14, 2023

Commit

eb0d2fc

1 Parent(s): 8d22ff5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -283,7 +283,7 @@ def gen_text(prompt, hf_model='meta-llama/Llama-2-70b-chat-hf', hf_token=None, p
 def gen_art(editor, cover_art_image, gen_cover_art_prompt):
   if gen_cover_art_prompt.strip() == "":
-    ppm = get_new_ppm(f"""describe the story below as a movie poster. give me the caption ONLY.
 --------------------------------
 {editor}""")
     cover_art_prompt = gen_text(ppm.build_prompts(), hf_model=MODEL_ID, hf_token=TOKEN)

 def gen_art(editor, cover_art_image, gen_cover_art_prompt):
   if gen_cover_art_prompt.strip() == "":
+    ppm = get_new_ppm(f"""describe the story below as a movie poster with ONLY general terms and adjectives. give me the caption ONLY.
 --------------------------------
 {editor}""")
     cover_art_prompt = gen_text(ppm.build_prompts(), hf_model=MODEL_ID, hf_token=TOKEN)