updated

app.py

@@ -3,8 +3,6 @@ from transformers import pipeline
 from PIL import Image
 
 
-
-
 st.header("Character Captions")
 st.write("Have a character caption any image you upload!")
 character = st.selectbox("Choose a character", ["rapper", "monkey", "shrek", "unintelligible"])
@@ -32,16 +30,5 @@ if uploaded_img is not None:
     prompt = character_prompts[character]
     st.write(prompt)
 
-
-
-
-# pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-Guard-2-8B")
-
-# # Pass the caption to another model
-
-# personality = "rapper"
-# prompt = personality_prompts[personality]
-
-# styled_caption = text_generator(prompt, max_length=100, num_return_sequences=1)
-
-# print(styled_caption[0]["generated_text"])
+    personality = "rapper"
+    prompt = character_prompts[personality]