added llama

app.py

@@ -21,7 +21,7 @@ if uploaded_img is not None:
     st.write("Caption:", caption)
 
     character_prompts = {
-        "rapper": f"Describe this scene like you're a rapper singing: {caption}.",
+        "rapper": f"Describe this scene like you're a rapper: {caption}.",
         "monkey": f"Describe this scene like you're a monkey going bananas: {caption}.",
         "shrek": f"Describe this scene like you're Shrek: {caption}.",
         "unintelligible": f"Describe this scene in a way that makes no sense: {caption}."
@@ -32,3 +32,14 @@ if uploaded_img is not None:
 
     personality = "rapper"
     prompt = character_prompts[personality]
+
+
+
+    text_generator = pipeline("text-generation", model="meta-llama/Llama-2-7b-hf")
+
+    prompt = character_prompts[character]
+    st.write("Styled Prompt:", prompt)  
+
+    generated_text = text_generator(prompt, max_length=50, do_sample=True)
+    styled_caption = generated_text[0]['generated_text']
+    st.write("Character-Styled Caption:", styled_caption)