updated

app.py

@@ -1,21 +1,16 @@
 import streamlit as st
 from transformers import pipeline
-from huggingface_hub import login, InferenceClient
+from huggingface_hub import InferenceClient
 from PIL import Image
 import os
 
 
-
-
-
-
-
-login(token=os.getenv("HUGGINGFACE_TOKEN"))
-client = InferenceClient(api_key="HUGGINGFACE_TOKEN")
+api_key = os.getenv("HUGGINGFACE_TOKEN")
+client = InferenceClient(api_key=api_key)
 
 st.header("Character Captions (IN PROGRESS!)")
 st.write("Have a character caption any image you upload!")
-character = st.selectbox("Choose a character", ["rapper", "monkey", "shrek", "unintelligible"])
+character = st.selectbox("Choose a character", ["rapper", "shrek", "unintelligible"])
 
 uploaded_img = st.file_uploader("Upload an image")
 
@@ -28,21 +23,14 @@ if uploaded_img is not None:
 
     response = image_captioner(image)
     caption = response[0]['generated_text']
-    st.write("Caption:", caption)
 
     character_prompts = {
         "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}."
     }
 
     prompt = character_prompts[character]
-    st.write(prompt)
-
-    personality = "rapper"
-    prompt = character_prompts[personality]
-
 
     messages = [
         { "role": "user", "content": prompt }
@@ -55,17 +43,11 @@ if uploaded_img is not None:
         stream=True
     )
 
+    response = ''
     for chunk in stream:
-        st.write(chunk.choices[0].delta.content)
-
-
-
+        response += chunk.choices[0].delta.content
+    
+    st.write(response)
 
-    # text_generator = pipeline("text-generation", model="meta-llama/Llama-2-7b-hf", framework="pt")
 
-    # 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)